Spaces:
Runtime error
Runtime error
Commit
•
b2f11c8
1
Parent(s):
2ee4f28
Upload folder using huggingface_hub
Browse files- Dockerfile +7 -0
Dockerfile
CHANGED
|
@@ -1,6 +1,13 @@
|
|
| 1 |
FROM python:3.9
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
ADD finetuning.py .
|
| 3 |
ADD finetuning_set_v1.0.json .
|
|
|
|
|
|
|
|
|
|
| 4 |
RUN pip install trl transformers
|
| 5 |
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 6 |
CMD ["python", "./finetuning.py"]
|
|
|
|
| 1 |
FROM python:3.9
|
| 2 |
+
|
| 3 |
+
RUN useradd -m -u 1000 user
|
| 4 |
+
USER user
|
| 5 |
+
|
| 6 |
ADD finetuning.py .
|
| 7 |
ADD finetuning_set_v1.0.json .
|
| 8 |
+
|
| 9 |
+
RUN chown -R user:user
|
| 10 |
+
|
| 11 |
RUN pip install trl transformers
|
| 12 |
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 13 |
CMD ["python", "./finetuning.py"]
|