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,
|