Update app.py

app.py

@@ -246,7 +246,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 if torch.cuda.is_available():
-    #vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
     #token = os.environ.get("HF_TOKEN")  # 从环境变量读取令牌
     model_path = hf_hub_download(
         repo_id="Menyu/ModelFile",  # 模型仓库名称（非完整URL）
@@ -255,7 +255,7 @@ if torch.cuda.is_available():
     )
     pipe = StableDiffusionXLPipeline.from_single_file(
         model_path,
-        #vae=vae, 内置VAE
+        vae=vae,
         use_safetensors=True,
         torch_dtype=torch.bfloat16,
     )
@@ -348,7 +348,7 @@ h1{text-align:center}
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成MiaoMiao Harem 1.6模型的图片""")
+        ### 快速生成NaixlMmmmix v50模型的图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(