Update app.py

app.py

@@ -28,7 +28,6 @@ for model_id, repo_id in all_model_id.items():
     
     pipes[model_id + '_ip_adapter'] = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
     pipes[model_id + '_ip_adapter'].load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
-    pipes[model_id + '_ip_adapter'].set_ip_adapter_scale(scale)
     LinFusion.construct_for(pipes[model_id + '_ip_adapter'])
     
     pipes[model_id + '_i2i'] = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
@@ -89,6 +88,7 @@ def infer_ip_adapter(model, prompt, image, scale, negative_prompt, seed, randomi
     generator = torch.Generator().manual_seed(seed)
 
     pipe = pipes[model + 'ip_adapter'].to(device)
+    pipe.set_ip_adapter_scale(scale)
     
     image = pipe(
         prompt = prompt,