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