pcalhoun commited on
Commit
8d4d042
·
verified ·
1 Parent(s): 0026caa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -129,11 +129,7 @@ def load_model_and_tokenizer():
129
  )
130
 
131
  # Load LoRA adapter if checkpoint exists
132
- if os.path.exists(CHECKPOINT_PATH):
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")