revert accidental change to load_model

zhvng · Mar 21, 2023 · a19337f · a19337f
1 parent c53e9f5
commit a19337f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/open_musiclm/config.py b/open_musiclm/config.py
@@ -188,7 +188,7 @@ def load_model(model, path):
     path = Path(path)
     assert path.exists(), f'checkpoint does not exist at {str(path)}'
     pkg = torch.load(str(path))
-    model.load_state_dict(pkg, strict=False)
+    model.load_state_dict(pkg)
 
 class disable_print:
     def __enter__(self):