Dockerfile

FROM python:3.9

WORKDIR /code

RUN useradd -m -u 1000 user
USER user

COPY finetuning.py /code/finetuning.py
COPY finetuning_set_v1.0.json /code/finetuning_set_v1.0.json

RUN mkdir -p /code && chmod 777 /code
# RUN chown -R user:user /code

RUN pip install trl transformers
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
CMD ["python", "./finetuning.py"]