Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -129,11 +129,7 @@ def load_model_and_tokenizer():
|
|
| 129 |
)
|
| 130 |
|
| 131 |
# Load LoRA adapter if checkpoint exists
|
| 132 |
-
|
| 133 |
-
model = PeftModel.from_pretrained(base_model, CHECKPOINT_PATH)
|
| 134 |
-
else:
|
| 135 |
-
print("Warning: Checkpoint path not found, using base model only")
|
| 136 |
-
model = base_model
|
| 137 |
|
| 138 |
model.eval()
|
| 139 |
print("✓ Model and LoRA adapter loaded successfully")
|
|
|
|
| 129 |
)
|
| 130 |
|
| 131 |
# Load LoRA adapter if checkpoint exists
|
| 132 |
+
model = PeftModel.from_pretrained(base_model, CHECKPOINT_PATH)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 133 |
|
| 134 |
model.eval()
|
| 135 |
print("✓ Model and LoRA adapter loaded successfully")
|