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, AutoencoderKL ,EulerDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, AutoencoderKL #,EulerDiscreteScheduler
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -14,14 +14,13 @@ 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",
+        "John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl",
         vae=vae,
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False
     )
-    
-    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: