shank commited on
Commit
9864e61
·
1 Parent(s): 6747185

Fix: Fixed again

Browse files
Files changed (1) hide show
  1. training/train_grpo.py +2 -2
training/train_grpo.py CHANGED
@@ -50,7 +50,7 @@ if not args.test_local:
50
  "accelerate==1.0.1",
51
  "trl==0.15.2",
52
  "peft==0.13.2",
53
- "bitsandbytes==0.44.1",
54
  ]
55
  print("Installing training dependencies...", flush=True)
56
  ret = os.system(
@@ -532,4 +532,4 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
532
  if HF_TOKEN and not args.test:
533
  model.push_to_hub(HF_REPO, token=HF_TOKEN, private=True)
534
  tokenizer.push_to_hub(HF_REPO, token=HF_TOKEN, private=True)
535
- print(f"Pushed to https://huggingface.co/{HF_REPO}")
 
50
  "accelerate==1.0.1",
51
  "trl==0.15.2",
52
  "peft==0.13.2",
53
+ "bitsandbytes==0.45.3",
54
  ]
55
  print("Installing training dependencies...", flush=True)
56
  ret = os.system(
 
532
  if HF_TOKEN and not args.test:
533
  model.push_to_hub(HF_REPO, token=HF_TOKEN, private=True)
534
  tokenizer.push_to_hub(HF_REPO, token=HF_TOKEN, private=True)
535
+ print(f"Pushed to https://huggingface.co/{HF_REPO}")