Upload ChessBot Chess model

README.md

@@ -44,22 +44,7 @@ model = model.to(device)
 
 # Get the best move
 move = model.get_move_from_fen_no_thinking(fen, T=0.1, device=device)
-print(f"Policy-based move: {move}")
-
-# Get the best move using value analysis
-value_move = model.get_best_move_value(fen, T=0, device=device)
-print(f"Value-based move: {value_move}")
-
-# Get position evaluation
-position_value = model.get_position_value(fen, device=device)
-print(f"Position value [black_win, draw, white_win]: {position_value}")
-
-# Get move probabilities
-probs = model.get_move_from_fen_no_thinking(fen, T=1, device=device, return_probs=True)
-top_moves = sorted(probs.items(), key=lambda x: x[1], reverse=True)[:5]
-print("Top 5 moves:")
-for move, prob in top_moves:
-    print(f"  {move}: {prob:.4f}")
+print(f"Predicted move: {move}")
 ```
 
 ## Requirements

config.json

@@ -3,6 +3,11 @@
   "d_model": 512,
   "max_position_embeddings": 64,
   "model_type": "chessbot",
+  "architectures": ["ChessBotModel"],
+  "auto_map": {
+    "AutoModel": "modeling_chessbot.ChessBotModel",
+    "AutoConfig": "modeling_chessbot.ChessBotConfig"
+  },
   "num_heads": 8,
   "num_layers": 10,
   "transformers_version": "4.53.1",