Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, EulerAncestralDiscreteScheduler
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -14,7 +14,7 @@ USE_TORCH_COMPILE = "1"
 ENABLE_CPU_OFFLOAD = "1"
 
 if torch.cuda.is_available():
-    pipe = StableDiffusionXLPipeline.from_pretrained(
+    pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl",
         torch_dtype=torch.float16,
         use_safetensors=True,