Make load_state_dict use strict=False

jbloomAus · Apr 20, 2024 · fdf7fe9 · fdf7fe9
1 parent 6a056b7
commit fdf7fe9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sae_lens/training/sparse_autoencoder.py b/sae_lens/training/sparse_autoencoder.py
@@ -312,7 +312,7 @@ def load_from_pretrained_legacy(cls, path: str):
 
         # Create an instance of the class using the loaded configuration
         instance = cls(cfg=state_dict["cfg"])
-        instance.load_state_dict(state_dict["state_dict"])
+        instance.load_state_dict(state_dict["state_dict"], strict=False)
 
         return instance