Dockerfile

FROM python:3.8-slim

RUN pip install --upgrade pip setuptools wheel
RUN pip install torch transformers datasets deepspeed vllm accelerate unsloth
RUN pip install huggingface_hub

WORKDIR /app
COPY . /app

CMD ["python", "train.py"]