Spaces:
Runtime error
Runtime error
phuong-d-h-nguyen
commited on
Commit
•
02ecb25
1
Parent(s):
a1c71bb
Upload folder using huggingface_hub
Browse files- finetuning.py +1 -1
finetuning.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from trl import SFTTrainer
|
2 |
import os
|
3 |
-
os.environ["
|
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
|