DiZH797 commited on
Commit
5afedcb
·
verified ·
1 Parent(s): 1a94036

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -58,6 +58,7 @@ def get_pipe(model_id: str, lora_scale: float = 1.0):
58
  pipe.load_lora_weights(
59
  LORA_MODEL_ID, weight_name="merged_lora_weights.safetensors"
60
  )
 
61
  else:
62
  # Load a standard model without LoRA
63
  pipe = DiffusionPipeline.from_pretrained(
 
58
  pipe.load_lora_weights(
59
  LORA_MODEL_ID, weight_name="merged_lora_weights.safetensors"
60
  )
61
+ pipe.fuse_lora(lora_scale=lora_scale)
62
  else:
63
  # Load a standard model without LoRA
64
  pipe = DiffusionPipeline.from_pretrained(