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/*

# Copy requirements
COPY requirements_v5.txt requirements.txt

# Install Python dependencies
# Note: Some packages may fail - that's OK, script has fallbacks
RUN pip install --no-cache-dir torch transformers accelerate trl datasets huggingface_hub peft bitsandbytes optimum || \
    pip install --no-cache-dir torch transformers accelerate trl datasets huggingface_hub peft optimum

# Try to install IPEX (may fail on non-Intel)
RUN pip install --no-cache-dir intel_extension_for_pytorch || echo "IPEX not installed (non-Intel CPU)"

# Copy training script
COPY train_arithmetic_v5_ultimate.py .

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

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