Nutnell commited on
Commit
255a9c8
·
verified ·
1 Parent(s): f179e52

Update fine_tune.py

Browse files
Files changed (1) hide show
  1. fine_tune.py +1 -1
fine_tune.py CHANGED
@@ -15,7 +15,7 @@ from trl import SFTTrainer
15
 
16
 
17
  base_model_name = "unsloth/llama-3-8b-Instruct-bnb-4bit"
18
- output_dir = "fine_tuning/results/llama-3-8b-instruct-direct-ed"
19
 
20
  dataset_path = "dataset.jsonl"
21
 
 
15
 
16
 
17
  base_model_name = "unsloth/llama-3-8b-Instruct-bnb-4bit"
18
+ output_dir = "/tmp/fine_tuning"
19
 
20
  dataset_path = "dataset.jsonl"
21