Spaces:

agentDebugger
/

AgentDebugger-training-v3

Running

shank commited on 12 days ago

Commit

6747185

1 Parent(s): 18b4e8a

Fix: Fixing Again

Files changed (1) hide show

training/train_grpo.py CHANGED Viewed

@@ -46,11 +46,11 @@ if not args.test_local:
     _TRAIN_DEPS = [
         "wandb==0.18.7",
         "datasets==3.0.2",
-        "transformers==4.44.2",
-        "accelerate==0.34.2",
         "trl==0.15.2",
         "peft==0.13.2",
-        "bitsandbytes==0.43.3",
     ]
     print("Installing training dependencies...", flush=True)
     ret = os.system(

     _TRAIN_DEPS = [
         "wandb==0.18.7",
         "datasets==3.0.2",
+        "transformers==4.48.3",
+        "accelerate==1.0.1",
         "trl==0.15.2",
         "peft==0.13.2",
+        "bitsandbytes==0.44.1",
     ]
     print("Installing training dependencies...", flush=True)
     ret = os.system(