File size: 790 Bytes
30bfe4c 204fa23 30bfe4c 204fa23 d1de63d 204fa23 54ad149 1a15203 204fa23 54ad149 1a15203 204fa23 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | FROM python:3.11-slim
WORKDIR /app
RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
# Stable cu121 torch — TorchAO import shims are applied at runtime before Unsloth loads.
RUN pip install --no-cache-dir \
torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
pip install --no-cache-dir \
trl transformers datasets accelerate peft bitsandbytes \
openenv-core fastapi uvicorn pydantic \
wandb faker python-dotenv gradio && \
pip install --no-cache-dir unsloth_zoo && \
pip install --no-cache-dir "unsloth @ git+https://github.com/unslothai/unsloth.git"
COPY . .
RUN python -m data.loader || echo "Will use fallback facts"
EXPOSE 7860
CMD ["python", "-m", "training.space_runner"]
|