rishabhsetiya commited on
Commit
dafbeb4
·
verified ·
1 Parent(s): bd9a4bb

Update fine_tuning.py

Browse files
Files changed (1) hide show
  1. 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