add vae

app.py

@@ -11,6 +11,10 @@ model_id = "segmind/Segmind-Vega"
 adapter_id = "segmind/Segmind-VegaRT"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = AutoPipelineForImage2Image.from_pretrained(model_id, torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained(model_id)
+pipe.vae = AutoencoderTiny.from_pretrained(
+    "madebyollin/taesd",
+    use_safetensors=True,
+)
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 
 pipe = pipe.to(device)