hirann commited on
Commit
776b1d2
·
verified ·
1 Parent(s): 0b7b124

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -1,9 +1,11 @@
1
  FROM unsloth/unsloth:latest
2
  WORKDIR /app
3
- # Pin transformers to a version known to work with Unsloth RL patching
4
- RUN /opt/venv/bin/python -m pip install --no-cache-dir -U huggingface_hub>=0.28.0 trl==0.15.0 transformers==4.47.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"]