Dockerfile

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