ViuAI commited on
Commit
f98eae7
·
verified ·
1 Parent(s): 29474cb

Upload runners/H200_TEST_SARUS_SFT.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. runners/H200_TEST_SARUS_SFT.txt +1 -1
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, max_seq_len=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))
 
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))