Upload folder using huggingface_hub

Dockerfile

@@ -8,7 +8,7 @@ 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 chown -R user:user /code
 
 RUN pip install trl transformers
 RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

finetuning.py

@@ -1,6 +1,6 @@
-from trl import SFTTrainer
 import os
 os.environ["HF_HOME"] = "/home/jonasdiez/neura_llm_kit/finetuning/cache"
+from trl import SFTTrainer
 from transformers import TrainingArguments
 import json
 import torch