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

chess-gpt-eval-contrastive/mamba_module.py

@@ -93,7 +93,7 @@ class MambaPlayer:
         self.hooks = []
         
         for i, layer in enumerate(self.model.backbone.layers):
-            self.activations[i] = {"won": [], "lost": []}
+            self.activations[i] = {"won": [], "lost": [], "current": []}
             
             def hook(module, input, output, layer_idx=i):
                 if isinstance(output, tuple):