Spaces:
Paused
Paused
Upload Dockerfile with huggingface_hub
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -1,9 +1,11 @@
|
|
| 1 |
FROM unsloth/unsloth:latest
|
| 2 |
WORKDIR /app
|
| 3 |
-
#
|
| 4 |
-
RUN /opt/venv/bin/python -m pip install --no-cache-dir -U huggingface_hub>=0.28.0 trl==0.15.0
|
| 5 |
COPY ./requirements.txt requirements.txt
|
| 6 |
RUN /opt/venv/bin/python -m pip install --no-cache-dir -r requirements.txt
|
| 7 |
COPY . /app
|
| 8 |
ENV PYTHONPATH=/app
|
|
|
|
|
|
|
| 9 |
ENTRYPOINT ["/opt/venv/bin/python", "training/launch_hf_training.py"]
|
|
|
|
| 1 |
FROM unsloth/unsloth:latest
|
| 2 |
WORKDIR /app
|
| 3 |
+
# Go back to official image versions, but ensure TRL and Hub are updated for GRPO
|
| 4 |
+
RUN /opt/venv/bin/python -m pip install --no-cache-dir -U huggingface_hub>=0.28.0 trl==0.15.0
|
| 5 |
COPY ./requirements.txt requirements.txt
|
| 6 |
RUN /opt/venv/bin/python -m pip install --no-cache-dir -r requirements.txt
|
| 7 |
COPY . /app
|
| 8 |
ENV PYTHONPATH=/app
|
| 9 |
+
# Disable VLLM to avoid the Gemma3Config dependency error
|
| 10 |
+
ENV UNSLOTH_USE_VLLM=0
|
| 11 |
ENTRYPOINT ["/opt/venv/bin/python", "training/launch_hf_training.py"]
|