shank commited on
Commit Β·
633a3b7
1
Parent(s): c945597
Align gradio version with Hugging Face Space builder2
Browse files- training/train_grpo.py +1 -1
training/train_grpo.py
CHANGED
|
@@ -44,7 +44,7 @@ if os.environ.get("FORCE_BOOTSTRAP_DEPS") == "1":
|
|
| 44 |
os.system(
|
| 45 |
f"{sys.executable} -m pip install -q "
|
| 46 |
"wandb==0.18.7 datasets==3.0.2 transformers==4.46.3 "
|
| 47 |
-
"accelerate==1.0.1 trl==0.
|
| 48 |
)
|
| 49 |
|
| 50 |
# ββ GPU/training imports (skipped in --test-local mode) βββββββββββββββββββββββ
|
|
|
|
| 44 |
os.system(
|
| 45 |
f"{sys.executable} -m pip install -q "
|
| 46 |
"wandb==0.18.7 datasets==3.0.2 transformers==4.46.3 "
|
| 47 |
+
"accelerate==1.0.1 trl==0.14.0 bitsandbytes==0.43.3 peft==0.13.2"
|
| 48 |
)
|
| 49 |
|
| 50 |
# ββ GPU/training imports (skipped in --test-local mode) βββββββββββββββββββββββ
|