File size: 549 Bytes
fd0c71a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
WORKDIR /app
ENV PYTHONUNBUFFERED=1 \
PIP_DISABLE_PIP_VERSION_CHECK=1 \
TOKENIZERS_PARALLELISM=false \
POLYGUARD_OFFLINE_MODE=false \
POLYGUARD_MODEL_ID=Qwen/Qwen2.5-0.5B-Instruct \
POLYGUARD_AUTORUN=1
COPY . .
RUN python -m pip install --upgrade pip setuptools wheel \
&& python -m pip install --no-cache-dir -r requirements.txt \
&& python -m pip install --no-cache-dir --no-build-isolation -e .
EXPOSE 7860
CMD ["python", "-m", "app.hf_space.training_runner"]
|