Spaces:
Paused
Paused
Update train_model.py
Browse files- 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=
|
| 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=
|
| 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,
|