Update fine_tuning.py
Browse files- fine_tuning.py +1 -0
fine_tuning.py
CHANGED
@@ -108,6 +108,7 @@ def preprocess(example):
|
|
108 |
# LOAD & TRAIN MODEL
|
109 |
# -----------------------------
|
110 |
def load_and_train(model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0"):
|
|
|
111 |
current_dir = os.path.dirname(os.path.abspath(__file__))
|
112 |
json_file_path = os.path.join(current_dir, 'makemytrip_qa_full.json')
|
113 |
|
|
|
108 |
# LOAD & TRAIN MODEL
|
109 |
# -----------------------------
|
110 |
def load_and_train(model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0"):
|
111 |
+
global model
|
112 |
current_dir = os.path.dirname(os.path.abspath(__file__))
|
113 |
json_file_path = os.path.join(current_dir, 'makemytrip_qa_full.json')
|
114 |
|