Spaces:
Sleeping
Sleeping
Fix: pin trl<0.17 for FSDP compat, skip world model (already done)
Browse files- 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"]
|