FROM python:3.11-slim

WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    && rm -rf /var/lib/apt/lists/*

# Install core Python packages ONLY - no IPEX, no OpenVINO
RUN pip install --no-cache-dir \
    torch \
    transformers \
    accelerate \
    trl \
    datasets \
    huggingface_hub \
    bitsandbytes

# Note: NO peft - we're testing without LoRA!

# Copy training script
COPY train_arithmetic_v9_no_lora.py .

# Set HF token from secret
ENV HF_TOKEN=${HF_TOKEN}

# Run training
CMD ["python", "train_arithmetic_v9_no_lora.py"]