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"
|
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)
|