Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import spaces
 import numpy as np
 import uuid
-from diffusers import PixArtAlphaPipeline, LCMScheduler, ConsistencyDecoderVAE
+from diffusers import PixArtAlphaPipeline, LCMScheduler
 import torch
 from typing import Tuple
 from datetime import datetime
@@ -80,8 +80,6 @@ pipe = PixArtAlphaPipeline.from_pretrained(
         torch_dtype=torch.float16,
         use_safetensors=True,
 ).to("cuda:0")
-
-pipe.vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=torch.float16)
 pipe.text_encoder.to_bettertransformer()
 
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]: