Upload runners/H200_TEST_SARUS_SFT.txt with huggingface_hub
Browse files
runners/H200_TEST_SARUS_SFT.txt
CHANGED
|
@@ -59,7 +59,7 @@ if not ckpt_path:
|
|
| 59 |
shutil.copy(dl, ckpt_path)
|
| 60 |
|
| 61 |
# 3. Load Model
|
| 62 |
-
args = ModelArgs(vocab_size=64003,
|
| 63 |
model = Transformer(args).to(device)
|
| 64 |
state = torch.load(ckpt_path, map_location=device, weights_only=False)
|
| 65 |
raw_sd = state.get("model", state.get("model_state_dict", state))
|
|
|
|
| 59 |
shutil.copy(dl, ckpt_path)
|
| 60 |
|
| 61 |
# 3. Load Model
|
| 62 |
+
args = ModelArgs(vocab_size=64003, context_length=2048)
|
| 63 |
model = Transformer(args).to(device)
|
| 64 |
state = torch.load(ckpt_path, map_location=device, weights_only=False)
|
| 65 |
raw_sd = state.get("model", state.get("model_state_dict", state))
|