phuong-d-h-nguyen commited on
Commit
8f0fca4
·
verified ·
1 Parent(s): 095ba3f

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -3
Dockerfile CHANGED
@@ -3,10 +3,13 @@ FROM python:3.9
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
 
3
  RUN useradd -m -u 1000 user
4
  USER user
5
 
6
+ WORKDIR /code
 
7
 
8
+
9
+ COPY finetuning.py /code/finetuning.py
10
+ COPY finetuning_set_v1.0.json /code/finetuning_set_v1.0.json
11
+
12
+ RUN chown -R user:user /code
13
 
14
  RUN pip install trl transformers
15
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu