Arijit-07 commited on
Commit
5e0397b
·
verified ·
1 Parent(s): d392552

Update train_model.py

Browse files
Files changed (1) hide show
  1. train_model.py +3 -3
train_model.py CHANGED
@@ -208,7 +208,7 @@ if resuming_from_checkpoint:
208
  model_name=checkpoint_path,
209
  max_seq_length=CONFIG['max_seq_length'],
210
  dtype=None,
211
- load_in_4bit=CONFIG['load_in_4bit'],
212
  )
213
  else:
214
  print("🆕 Starting fresh model...")
@@ -216,10 +216,10 @@ else:
216
  model_name=CONFIG['model_name'],
217
  max_seq_length=CONFIG['max_seq_length'],
218
  dtype=None,
219
- load_in_4bit=CONFIG['load_in_4bit'],
220
  token=HF_TOKEN,
221
  )
222
-
223
  if not resuming_from_checkpoint:
224
  model = FastLanguageModel.get_peft_model(
225
  model,
 
208
  model_name=checkpoint_path,
209
  max_seq_length=CONFIG['max_seq_length'],
210
  dtype=None,
211
+ load_in_4bit=False, # 🔥 FORCE DISABLE
212
  )
213
  else:
214
  print("🆕 Starting fresh model...")
 
216
  model_name=CONFIG['model_name'],
217
  max_seq_length=CONFIG['max_seq_length'],
218
  dtype=None,
219
+ load_in_4bit=False, # 🔥 FORCE DISABLE
220
  token=HF_TOKEN,
221
  )
222
+
223
  if not resuming_from_checkpoint:
224
  model = FastLanguageModel.get_peft_model(
225
  model,