HaileyStorm commited on
Commit
1401d32
1 Parent(s): c4093d1

Update chess-gpt-eval/main.py

Browse files

Properly instantiate models based on move_num_in_gamestate (vs hard-coded value)

Files changed (1) hide show
  1. chess-gpt-eval/main.py +2 -2
chess-gpt-eval/main.py CHANGED
@@ -550,9 +550,9 @@ if __name__ == "__main__":
550
  # player_one = StockfishPlayer(skill_level=-1, play_time=0.1)
551
 
552
  player_one_recording_name = nanogpt_player
553
- player_one = NanoGptPlayer(model_name=player_one_recording_name, move_num_in_gamestate=False)
554
  player_one_recording_name = "xformer_" + nanogpt_player
555
- #player_one = MambaPlayer(model_name=player_one_recording_name, move_num_in_gamestate=False)
556
  player_two = StockfishPlayer(skill_level=i, play_time=0.1)
557
  #player_two = LC0PLayer(skill=i)
558
 
 
550
  # player_one = StockfishPlayer(skill_level=-1, play_time=0.1)
551
 
552
  player_one_recording_name = nanogpt_player
553
+ player_one = NanoGptPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
554
  player_one_recording_name = "xformer_" + nanogpt_player
555
+ #player_one = MambaPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
556
  player_two = StockfishPlayer(skill_level=i, play_time=0.1)
557
  #player_two = LC0PLayer(skill=i)
558