Rayugacodes commited on
Commit
f4c4a2c
·
verified ·
1 Parent(s): b3f08a7

Fix: pin trl<0.17 for FSDP compat, skip world model (already done)

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -21,7 +21,7 @@ RUN pip install --no-cache-dir \
21
 
22
  RUN pip install --no-cache-dir \
23
  transformers \
24
- trl \
25
  peft \
26
  datasets \
27
  accelerate \
@@ -31,4 +31,4 @@ COPY train_on_hf.py .
31
  RUN chmod -R 777 /app
32
 
33
  EXPOSE 7860
34
- CMD ["sh", "-c", "python3 train_on_hf.py --hf-token $HF_TOKEN"]
 
21
 
22
  RUN pip install --no-cache-dir \
23
  transformers \
24
+ "trl>=0.12,<0.17" \
25
  peft \
26
  datasets \
27
  accelerate \
 
31
  RUN chmod -R 777 /app
32
 
33
  EXPOSE 7860
34
+ CMD ["sh", "-c", "python3 train_on_hf.py --hf-token $HF_TOKEN --skip-world-model"]