add model

app.py

@@ -3,13 +3,19 @@ import modin.pandas as pd
 import torch
 import numpy as np
 from PIL import Image
-from diffusers import AutoPipelineForImage2Image
+from diffusers import LCMScheduler, AutoPipelineForImage2Image
 from diffusers.utils import load_image
 import math
 
+model_id = "segmind/Segmind-Vega"
+adapter_id = "segmind/Segmind-VegaRT"
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
+pipe = AutoPipelineForImage2Image.from_pretrained(model_id, torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained(model_id)
+pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
+
 pipe = pipe.to(device)
+pipe.load_lora_weights(adapter_id)
+pipe.fuse_lora()
 
 def resize(value,img):
     img = Image.open(img)