Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|