phuong-d-h-nguyen commited on
Commit
02ecb25
1 Parent(s): a1c71bb

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. finetuning.py +1 -1
finetuning.py CHANGED
@@ -1,6 +1,6 @@
1
  from trl import SFTTrainer
2
  import os
3
- os.environ["TRANSFORMERS_CACHE"] = "/home/jonasdiez/neura_llm_kit/finetuning/cache"
4
  from transformers import TrainingArguments
5
  import json
6
  import torch
 
1
  from trl import SFTTrainer
2
  import os
3
+ os.environ["HF_HOME"] = "/home/jonasdiez/neura_llm_kit/finetuning/cache"
4
  from transformers import TrainingArguments
5
  import json
6
  import torch