Spaces:
Running
Running
Add vllm to dependencies to fix TRL's hard import requirement
Browse files
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",
|