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

chess-gpt-eval-contrastive/mamba_module.py

@@ -94,7 +94,7 @@ class MambaPlayer:
             self.activations[i] = {"won": [], "lost": []}
             
             def hook(module, input, output, layer_idx=i):
-                self.activations[layer_idx]["current"] = output.detach().cpu().numpy()
+                self.activations[layer_idx]["current"] = output#.detach().cpu().numpy()
             
             self.hooks.append(layer.register_forward_hook(hook))