Upload chess-gpt-eval/nanogpt/nanogpt_module.py with huggingface_hub

chess-gpt-eval/nanogpt/nanogpt_module.py

@@ -95,7 +95,7 @@ class NanoGptPlayer:
             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.encode = encode
         self.decode = decode
@@ -134,11 +134,14 @@ class NanoGptPlayer:
         return model_response
 
     def get_move_from_response(self, response: str) -> str:
-        # Parse the response to get only the first move
-        moves = response.split()
-        first_move = moves[0]
+        try:
+            # Parse the response to get only the first move
+            moves = response.split()
+            first_move = moves[0]
+            return first_move
+        except:
+            return None
 
-        return first_move
 
     def get_move(self, board: str, game_state: str, temperature: float) -> str:
         completion = self.get_nanogpt_response(game_state, temperature)