pin vllm<=0.18.0 to fix TRL importance-sampling ratio collapse
Browse files- pyproject.toml +5 -1
pyproject.toml
CHANGED
|
@@ -38,7 +38,11 @@ train = [
|
|
| 38 |
"transformers>=4.46.0",
|
| 39 |
"datasets>=2.20.0",
|
| 40 |
"accelerate>=1.0.0",
|
| 41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
"trackio",
|
| 43 |
"torch>=2.4.0",
|
| 44 |
# Local plotting of loss + reward curves (hackathon submission evidence).
|
|
|
|
| 38 |
"transformers>=4.46.0",
|
| 39 |
"datasets>=2.20.0",
|
| 40 |
"accelerate>=1.0.0",
|
| 41 |
+
# Pin to TRL's supported window. vllm 0.19+ changes the sampling-logprob
|
| 42 |
+
# path and breaks GRPO importance sampling (ratio collapses to ~0,
|
| 43 |
+
# grad_norm goes to 0, model stops learning). Keep <=0.18.0 until TRL
|
| 44 |
+
# bumps its supported range.
|
| 45 |
+
"vllm>=0.11.0,<=0.18.0",
|
| 46 |
"trackio",
|
| 47 |
"torch>=2.4.0",
|
| 48 |
# Local plotting of loss + reward curves (hackathon submission evidence).
|