phuong-d-h-nguyen commited on
Commit
2ee4f28
1 Parent(s): 0dcea97

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. finetuning.py +3 -0
finetuning.py CHANGED
@@ -2,6 +2,9 @@ from trl import SFTTrainer
2
  from transformers import TrainingArguments
3
  import json
4
  import torch
 
 
 
5
 
6
  with open("finetuning_set_v1.0.json") as f:
7
  dataset = json.load(f)
 
2
  from transformers import TrainingArguments
3
  import json
4
  import torch
5
+ import os
6
+
7
+ os.environ["TRANSFORMERS_CACHE"] = "./cache"
8
 
9
  with open("finetuning_set_v1.0.json") as f:
10
  dataset = json.load(f)