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,