Nutnell commited on
Commit
ee81f35
·
verified ·
1 Parent(s): be670a7

Update fine_tune.py

Browse files
Files changed (1) hide show
  1. fine_tune.py +1 -1
fine_tune.py CHANGED
@@ -14,7 +14,7 @@ from fastapi import FastAPI
14
  import uvicorn
15
 
16
  base_model_name = "unsloth/llama-3-8b-Instruct-bnb-4bit"
17
- output_dir = "fine_tuning" # persist inside /code
18
 
19
  dataset_path = "dataset.jsonl"
20
 
 
14
  import uvicorn
15
 
16
  base_model_name = "unsloth/llama-3-8b-Instruct-bnb-4bit"
17
+ output_dir = "/data/fine_tuning" # persist inside /code
18
 
19
  dataset_path = "dataset.jsonl"
20