Spaces:
Runtime error
Runtime error
File size: 293 Bytes
9b148bc b2f11c8 9b148bc b2f11c8 2414d77 9b148bc |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
FROM python:3.9
RUN useradd -m -u 1000 user
USER user
ADD finetuning.py .
ADD finetuning_set_v1.0.json .
RUN chown -R user:user
RUN pip install trl transformers
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
CMD ["python", "./finetuning.py"] |