| FROM python:3.10-slim | |
| WORKDIR /code | |
| # Install system dependencies | |
| # build-essential is often needed for compiling extensions | |
| RUN apt-get update && apt-get install -y \ | |
| libsndfile1 \ | |
| git \ | |
| build-essential \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Copy requirements first to leverage caching | |
| COPY requirements.txt . | |
| # Install Python dependencies | |
| # This handles torch, transformers, accelerate, etc. | |
| RUN pip install --no-cache-dir --upgrade pip && \ | |
| pip install --no-cache-dir -r requirements.txt | |
| COPY . . | |
| # Run training configuration | |
| CMD ["python", "-m", "src.train_hf"] | |