File size: 139 Bytes
9b148bc
 
 
 
 
1
2
3
4
5
FROM python:3.9
ADD finetuning.py .
ADD finetuning_set_v1.0.json .
RUN pip install trl transformers torch
CMD ["python", "./finetuning.py"]