Update app.py

app.py

@@ -5,6 +5,8 @@ import random
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 import torch
 
+from src.linfusion import LinFusion
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 all_model_id = {
  "DreamShaper-8": "Lykon/dreamshaper-8",
@@ -30,6 +32,7 @@ def infer_t2i(model, prompt, negative_prompt, seed, randomize_seed, width, heigh
 
  pipe = StableDiffusionPipeline.from_pretrained(all_model_id[model], torch_dtype=torch_dtype)
  pipe = pipe.to(device)
+ linfusion = LinFusion.construct_for(pipe)
  
  image = pipe(
  prompt = prompt, 
@@ -53,6 +56,7 @@ def infer_i2i(model, prompt, image, strength, negative_prompt, seed, randomize_s
 
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(all_model_id[model], torch_dtype=torch_dtype)
  pipe = pipe.to(device)
+ linfusion = LinFusion.construct_for(pipe)
  
  image = pipe(
  prompt = prompt, 
@@ -80,6 +84,7 @@ def infer_ip_adapter(model, prompt, image, scale, negative_prompt, seed, randomi
  pipe = pipe.to(device)
  pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter-plus_sd15.bin")
  pipeline.set_ip_adapter_scale(scale)
+ linfusion = LinFusion.construct_for(pipe)
  
  image = pipe(
  prompt = prompt,