shank commited on
Commit
633a3b7
Β·
1 Parent(s): c945597

Align gradio version with Hugging Face Space builder2

Browse files
Files changed (1) hide show
  1. 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.12.2 bitsandbytes==0.43.3 peft==0.13.2"
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) ───────────────────────