File size: 790 Bytes
77d8bcf
 
 
 
 
 
a262689
77d8bcf
98317c2
 
dee46e3
77d8bcf
98317c2
 
dee46e3
77d8bcf
 
 
 
 
 
 
 
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"]