shank commited on
Commit ·
accb271
1
Parent(s): 9864e61
Fix: Fixed again again
Browse files- training/train_grpo.py +1 -1
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.45.
|
| 54 |
]
|
| 55 |
print("Installing training dependencies...", flush=True)
|
| 56 |
ret = os.system(
|
|
|
|
| 50 |
"accelerate==1.0.1",
|
| 51 |
"trl==0.15.2",
|
| 52 |
"peft==0.13.2",
|
| 53 |
+
"bitsandbytes==0.45.5",
|
| 54 |
]
|
| 55 |
print("Installing training dependencies...", flush=True)
|
| 56 |
ret = os.system(
|