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,