Just in case to prevent resource leakage. (#5)

- Just in case to prevent resource leakage. (2f8eea64376276362f9fdc83748a4194ef07a660)


Co-authored-by: John Smith <[email protected]>

app.py

@@ -351,6 +351,8 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
                 low_cpu_mem_usage=True,
                 adapter_name=lora_name
             )
+            if image_input is not None: pipe_i2i = pipe_to_use
+            else: pipe = pipe_to_use
         print("Loaded LoRAs:", lora_names)
         print("Adapter weights:", lora_weights)
         if image_input is not None: