Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-from diffusers import AutoPipelineForText2Image, EulerDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, AutoencoderKL #,EulerDiscreteScheduler
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -12,13 +12,17 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096
 
 if torch.cuda.is_available():
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     pipe = AutoPipelineForText2Image.from_pretrained(
         "WhiteAiZ/noobaiXLNAIXL_epsilonPred075_diffusers",
+        vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False
+        add_watermarker=False,
+        variant="fp16"
     )
-    pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
+    
+    # pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: