Upload chess-gpt-eval/mamba_module.py with huggingface_hub

chess-gpt-eval/mamba_module.py

@@ -76,7 +76,7 @@ class MambaPlayer:
             vocab_size = len(stoi)
             print(f"Vocab size {vocab_size}")
             encode = lambda s: [stoi[c] for c in s.replace('-', '')]
-            decode = lambda l: "".join([itos[i] for i in l]).replace("OOO", "O-O-O").replace("OO", "O-O")
+            decode = lambda l: "".join([itos[i] for i in l if i < vocab_size]).replace("OOO", "O-O-O").replace("OO", "O-O")
 
         self.vocab_size = vocab_size
         self.encode = encode