Spaces:
Runtime error
Runtime error
File size: 404 Bytes
9b148bc b2f11c8 3d1c3e3 8f0fca4 b2f11c8 8fbcc74 8f0fca4 3d1c3e3 b2f11c8 2414d77 8fbcc74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
FROM python:3.9
RUN mkdir -p /code && chmod 777 /code
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 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"]
|