Update app.py

app.py

@@ -82,11 +82,7 @@ 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()
-            
+        joint_attention_kwargs = {"scale": 1}    
         if orginal_image:
             generated_image = img2img(
                 prompt=prompt,
@@ -110,8 +106,7 @@ def generate_image(orginal_image, prompt, adapter_names,  steps, seed, image_str
                 generator=generator,
                 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
 
@@ -171,10 +166,6 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
     # Load LoRA weights
     gr.Info("Start to load LoRA ...")
     
-    with calculateDuration("Unloading LoRA"):
-        pipe.unload_lora_weights()
-        img2img.unload_lora_weights()
-    
     lora_configs = None
     adapter_names = []
     
@@ -193,25 +184,17 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
                     weights = lora_info.get("weights")
                     adapter_name = lora_info.get("adapter_name")
                     adapter_weight = lora_info.get("adapter_weight")
-                    
+                    adapter_names.append(adapter_name)
+                    adapter_weights.append(adapter_weight)
                     if lora_repo and weights and adapter_name:
-                        retry_count = 3
-                        for attempt in range(retry_count):
-                            try:
-                                if img2img_model:
-                                    img2img.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
-                                else:
-                                    pipe.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
-                                adapter_names.append(adapter_name)
-                                adapter_weights.append(adapter_weight)
-                                break  # Load successful, exit retry loop
-                            except ValueError as e:
-                                print(f"Attempt {attempt+1}/{retry_count} failed to load LoRA adapter: {e}")
-                                if attempt == retry_count - 1:
-                                    print(f"Error loading LoRA adapter: {adapter_name} after {retry_count} attempts")
-                                else:
-                                    time.sleep(1)  # Wait before retrying
-                        
+                        try:
+                            if img2img_model:
+                                img2img.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
+                            else:
+                                pipe.load_lora_weights(lora_repo, weight_name=weights, adapter_name=adapter_name)
+                        except ValueError as e:
+                            print(f"Attempt {attempt+1}/{retry_count} failed to load LoRA adapter: {e}")
+                
                 # set lora weights
                 if len(adapter_names) > 0:
                     if img2img_model: