| FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime | |
| RUN apt-get update && \ | |
| apt-get install -y stockfish && \ | |
| rm -rf /var/lib/apt/lists/* | |
| WORKDIR /app | |
| COPY requirements_space.txt . | |
| RUN pip install --no-cache-dir -r requirements_space.txt | |
| COPY train_hf.py . | |
| COPY finetune_hf.py . | |
| COPY src/ src/ | |
| CMD ["python", "train_hf.py"] | |