Neon-AI commited on
Commit
9076aee
·
verified ·
1 Parent(s): 3626da7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -33,13 +33,14 @@ def load_model():
33
  )
34
 
35
  # Load LoRA from /lora if it exists
36
- try:
37
- model = PeftModel.from_pretrained(
38
- base_model,
39
- f"{MODEL_ID}/lora"
40
- )
41
- except Exception:
42
- model = base_model
 
43
 
44
  model.to("cpu")
45
  model.eval()
 
33
  )
34
 
35
  # Load LoRA from /lora if it exists
36
+ model = PeftModel.from_pretrained(
37
+ base_model,
38
+ f"{MODEL_ID}/lora",
39
+ is_trainable=False
40
+ )
41
+
42
+ print("LoRA loaded")
43
+ print(model.peft_config)
44
 
45
  model.to("cpu")
46
  model.eval()