Update chess-gpt-eval-contrastive/mamba_module.py

chess-gpt-eval-contrastive/mamba_module.py

@@ -189,10 +189,10 @@ class MambaPlayer:
                 "lost": len(layer_activations["lost"])
             }
     
-    with open(path, "wb") as f:
-        pickle.dump((activations_sum, activations_count), f)
+        with open(path, "wb") as f:
+            pickle.dump((activations_sum, activations_count), f)
     
-    self.activations = {}
+        self.activations = {}
     
     def apply_contrastive_activations(self, path):
         if os.path.exists(path):