Update app.py
Browse files
app.py
CHANGED
|
@@ -516,7 +516,7 @@ def run_lora(prompt, cfg_scale, steps, selected_indices, lora_scale_1, lora_scal
|
|
| 516 |
lora_path = lora['repo']
|
| 517 |
weight_name = lora.get("weights")
|
| 518 |
print(f"Lora Path: {lora_path}")
|
| 519 |
-
|
| 520 |
pipe_to_use.load_lora_weights(
|
| 521 |
lora_path,
|
| 522 |
weight_name=weight_name if weight_name else None,
|
|
|
|
| 516 |
lora_path = lora['repo']
|
| 517 |
weight_name = lora.get("weights")
|
| 518 |
print(f"Lora Path: {lora_path}")
|
| 519 |
+
pipe_to_use = pipe_i2i if image_input is not None else pipe
|
| 520 |
pipe_to_use.load_lora_weights(
|
| 521 |
lora_path,
|
| 522 |
weight_name=weight_name if weight_name else None,
|