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 StableDiffusionXLPipeline, AutoencoderKL, EulerDiscreteScheduler
 from compel import Compel, ReturnedEmbeddingsType
 
 import re
@@ -225,12 +225,12 @@ 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(
-        "Menyu/noobaiXLNAIXL_vPred05Version",
+    vae = StableDiffusionXLPipeline.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",
         vae=vae,
-        torch_dtype=torch.float16,
         use_safetensors=True,
+        torch_dtype=torch.float16,
         add_watermarker=False
     )
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
@@ -306,7 +306,7 @@ footer {
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成NoobAIXL V预测版本的模型图片""")
+        ### 快速生成NoobAIXL V预测0.6版本的模型图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(