Update app.py

app.py

@@ -143,7 +143,7 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
            
             with calculateDuration("Loading LoRA weights"):
                 adapter_weights = []
-                txt2img_pipe.to(device)
+                
                 for idx, lora_info in enumerate(lora_configs):
                     lora_repo = lora_info.get("repo")
                     weights = lora_info.get("weights")
@@ -156,12 +156,14 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
                     adapter_weights.append(adapter_weight)
                     if lora_repo and weights and adapter_name:
                         try:
+                            txt2img_pipe.to(device)
                             txt2img_pipe.load_lora_weights(lora_repo, weight_name=weights, adapter_name=lora_name)
                         except:
                             print("load lora error")
                 
                 # set lora weights
                 if len(lora_names) > 0:
+                    txt2img_pipe.to(device)
                     txt2img_pipe.set_adapters(lora_names, adapter_weights=adapter_weights)
     
     # Generate image