HaileyStorm commited on
Commit
17dd4f5
1 Parent(s): 418a88f

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

Browse files
chess-gpt-eval/nanogpt/nanogpt_module.py CHANGED
@@ -57,7 +57,7 @@ class NanoGptPlayer:
57
  if init_from == "resume":
58
  # init from a model saved in a specific directory
59
  #ckpt_path = os.path.join(BASE_DIR, out_dir, self.model_name)
60
- ckpt_path = os.path.normpath(f"../../mamba.py/out/{self.model_name}")
61
  checkpoint = torch.load(ckpt_path, map_location=device)
62
  #gptconf = GPTConfig(**checkpoint["model_args"])
63
  #model = GPT(gptconf)
 
57
  if init_from == "resume":
58
  # init from a model saved in a specific directory
59
  #ckpt_path = os.path.join(BASE_DIR, out_dir, self.model_name)
60
+ ckpt_path = os.path.normpath(f"../chess-mamba-vs-xformer/out/Xformer/{self.model_name}")
61
  checkpoint = torch.load(ckpt_path, map_location=device)
62
  #gptconf = GPTConfig(**checkpoint["model_args"])
63
  #model = GPT(gptconf)