Rayugacodes commited on
Commit
e191204
·
verified ·
1 Parent(s): 9ff82e1

Fix: trl==0.15.2 (has GRPO, no vllm/FSDP dep)

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -7
Dockerfile CHANGED
@@ -20,16 +20,17 @@ RUN pip install --no-cache-dir \
20
  torch --index-url https://download.pytorch.org/whl/cu121
21
 
22
  RUN pip install --no-cache-dir \
23
- "transformers==4.46.3" \
24
- "trl==0.12.2" \
25
- "peft==0.13.2" \
26
  "datasets>=2.18" \
27
- "accelerate==0.34.2" \
28
  huggingface_hub
29
 
30
- # Verify imports work
31
- RUN python3 -c "from trl import SFTTrainer, SFTConfig, GRPOConfig, GRPOTrainer; print('TRL imports OK')"
32
- RUN python3 -c "import torch; print('CUDA available:', torch.cuda.is_available())"
 
33
 
34
  COPY train_on_hf.py .
35
  RUN chmod -R 777 /app
 
20
  torch --index-url https://download.pytorch.org/whl/cu121
21
 
22
  RUN pip install --no-cache-dir \
23
+ "transformers>=4.46,<4.48" \
24
+ "trl==0.15.2" \
25
+ "peft>=0.13,<0.15" \
26
  "datasets>=2.18" \
27
+ "accelerate>=0.34,<0.36" \
28
  huggingface_hub
29
 
30
+ # Verify all imports work at build time
31
+ RUN python3 -c "from trl import SFTTrainer, SFTConfig, GRPOConfig, GRPOTrainer; print('TRL OK')"
32
+ RUN python3 -c "from peft import LoraConfig; print('PEFT OK')"
33
+ RUN python3 -c "import torch; print('CUDA:', torch.cuda.is_available())"
34
 
35
  COPY train_on_hf.py .
36
  RUN chmod -R 777 /app