Update app.py

app.py

@@ -82,6 +82,11 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
     
     with calculateDuration("Generating image"):
         # Generate image
+        joint_attention_kwargs = {"scale": 1}
+        if len(adapter_names) == 0:
+            joint_attention_kwargs = None
+            pipe.unload_lora_weights()
+            
         if orginal_image:
             generated_image = img2img(
                 prompt=prompt,
@@ -92,7 +97,7 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
                 width=width,
                 height=height,
                 generator=generator,
-                joint_attention_kwargs={"scale": 1}
+                joint_attention_kwargs=joint_attention_kwargs
             ).images[0]
         else:
             generated_image = pipe(
@@ -103,9 +108,10 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
                 height=height,
                 max_sequence_length=512,
                 generator=generator,
-                joint_attention_kwargs={"scale": 1}
+                joint_attention_kwargs=joint_attention_kwargs
             ).images[0]
-    
+    if len(adapter_names) > 0:
+        pipe.unload_lora_weights()
     progress(99, "Generate image success!")
     return generated_image