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, AutoencoderKL, EulerDiscreteScheduler
+from diffusers import AutoPipelineForText2Image, AutoencoderKL, EulerDiscreteScheduler
 from compel import Compel, ReturnedEmbeddingsType
 
 import re
@@ -226,11 +226,11 @@ MAX_IMAGE_SIZE = 4096
 
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-    pipe = StableDiffusionXLPipeline.from_single_file(
-        "https://huggingface.co/Laxhar/noobai-XL-Vpred-0.6/blob/main/noobai-xl-vpred-v0.6.safetensors",
+    pipe = AutoPipelineForText2Image.from_pretrained(
+        "Menyu/noobai-xl-vpred-v0_6",
         vae=vae,
-        use_safetensors=True,
         torch_dtype=torch.float16,
+        use_safetensors=True,
         add_watermarker=False
     )
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
@@ -356,9 +356,9 @@ with gr.Blocks(css=css) as demo:
             guidance_scale = gr.Slider(
                 label="Guidance Scale",
                 minimum=0.1,
-                maximum=10,
+                maximum=8,
                 step=0.1,
-                value=5.0,
+                value=3.0,
             )
             num_inference_steps = gr.Slider(
                 label="生成步数",