9b148bc 2414d77 9b148bc
1
2
3
4
5
6
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"]
