shank commited on
Commit ·
9864e61
1
Parent(s): 6747185
Fix: Fixed again
Browse files- 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.
|
| 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}")
|