hard007ik commited on
Commit
8464e70
·
1 Parent(s): 1ecce94

pin vllm<=0.18.0 to fix TRL importance-sampling ratio collapse

Browse files
Files changed (1) hide show
  1. 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
- "vllm>=0.6.3",
 
 
 
 
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).