Update app.py
Browse files
app.py
CHANGED
@@ -517,8 +517,8 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
|
|
517 |
low_cpu_mem_usage=True,
|
518 |
adapter_name=lora_name
|
519 |
)
|
520 |
-
|
521 |
-
|
522 |
print("Loaded LoRAs:", lora_names)
|
523 |
print("Adapter weights:", lora_weights)
|
524 |
if image_input is not None:
|
|
|
517 |
low_cpu_mem_usage=True,
|
518 |
adapter_name=lora_name
|
519 |
)
|
520 |
+
if image_input is not None: pipe_i2i = pipe_to_use
|
521 |
+
else: pipe = pipe_to_use
|
522 |
print("Loaded LoRAs:", lora_names)
|
523 |
print("Adapter weights:", lora_weights)
|
524 |
if image_input is not None:
|