phuong-d-h-nguyen commited on
Commit
8ce7345
1 Parent(s): 2fb4973

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. finetuning.py +1 -1
Dockerfile CHANGED
@@ -8,7 +8,7 @@ USER user
8
  COPY finetuning.py /code/finetuning.py
9
  COPY finetuning_set_v1.0.json /code/finetuning_set_v1.0.json
10
 
11
- RUN mkdir -p /code && chmod 777 /code
12
  # RUN chown -R user:user /code
13
 
14
  RUN pip install trl transformers
 
8
  COPY finetuning.py /code/finetuning.py
9
  COPY finetuning_set_v1.0.json /code/finetuning_set_v1.0.json
10
 
11
+ # RUN mkdir -p /code && chmod 777 /code
12
  # RUN chown -R user:user /code
13
 
14
  RUN pip install trl transformers
finetuning.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- os.environ["HF_HOME"] = "/home/jonasdiez/neura_llm_kit/finetuning/cache"
3
  from trl import SFTTrainer
4
  from transformers import TrainingArguments
5
  import json
 
1
  import os
2
+ os.environ["HF_HOME"] = "./code"
3
  from trl import SFTTrainer
4
  from transformers import TrainingArguments
5
  import json