Update model architecture: d_ff=1024, new weights from merged7.pt

modeling_chessbot.py

@@ -626,7 +626,7 @@ class ChessBotModel(ChessBotPreTrainedModel):
             probs = F.softmax(logits/T, dim=-1)
             sampled = torch.multinomial(probs, num_samples=1)
             if return_probs:
-                return sampled, probs
+                return probs
         
         move = policy_index[sampled.item()]
         return move