md896 commited on
Commit
397face
·
1 Parent(s): 16dd181

Downgrade TRL to <0.24.0 to fix missing dependency chain

Browse files
Files changed (1) hide show
  1. ultimate_sota_training.py +2 -1
ultimate_sota_training.py CHANGED
@@ -66,9 +66,10 @@ def bootstrap_deps() -> None:
66
  "--break-system-packages",
67
  "httpx>=0.27.0",
68
  "datasets>=3.4.1,<4.4.0",
69
- "trl>=0.18.2,<=0.24.0",
70
  "mergekit",
71
  "llm-blender",
 
72
  "wandb",
73
  "matplotlib",
74
  ]
 
66
  "--break-system-packages",
67
  "httpx>=0.27.0",
68
  "datasets>=3.4.1,<4.4.0",
69
+ "trl>=0.18.2,<0.24.0",
70
  "mergekit",
71
  "llm-blender",
72
+ "weave",
73
  "wandb",
74
  "matplotlib",
75
  ]