md896 commited on
Commit
711ae38
·
1 Parent(s): 97cddc4

Add vllm to dependencies to fix TRL's hard import requirement

Browse files
Files changed (1) hide show
  1. ultimate_sota_training.py +1 -0
ultimate_sota_training.py CHANGED
@@ -69,6 +69,7 @@ def bootstrap_deps() -> None:
69
  "trl>=0.18.2,<0.24.0",
70
  "mergekit",
71
  "llm-blender",
 
72
  "weave",
73
  "wandb",
74
  "matplotlib",
 
69
  "trl>=0.18.2,<0.24.0",
70
  "mergekit",
71
  "llm-blender",
72
+ "vllm",
73
  "weave",
74
  "wandb",
75
  "matplotlib",