Update app.py

app.py

@@ -249,8 +249,8 @@ if torch.cuda.is_available():
     # 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/Pixel",  # 模型仓库名称（非完整URL）
-        filename="MiaoMiaoPixel_V1.0.safetensors",
+        repo_id="Menyu/miaomiaoHarem_vPredDogma10",  # 模型仓库名称（非完整URL）
+        filename="miaomiaoHarem_vPredDogma10.safetensors",
         use_auth_token=token
     )
     pipe = StableDiffusionXLPipeline.from_single_file(
@@ -260,6 +260,10 @@ if torch.cuda.is_available():
         torch_dtype=torch.float16,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    pipe.scheduler.register_to_config(
+        prediction_type="v_prediction",
+        rescale_betas_zero_snr=True,
+    )
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -324,9 +328,9 @@ def infer(
         "height": height,
         "guidance_scale": guidance_scale,
         "num_inference_steps": num_inference_steps,
-        "model": "MiaoMiaoPixel_V1.0",
+        "model": "miaomiaoHarem_vPredDogma10",
         "use_resolution_binning": use_resolution_binning,
-        "PreUrl": "https://huggingface.co/spaces/Menyu/MiaoPixelNew"
+        "PreUrl": "https://huggingface.co/spaces/Menyu/miaomiaoHaremDogma"
     }
     # Add metadata to the image
     image_with_metadata = add_metadata_to_image(image, metadata)
@@ -345,7 +349,7 @@ h1{text-align:center}
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
-        ### 快速生成MiaoMiaoPixel 像素模型的图片""")
+        ### 快速生成 MiaomiaoHarem vPred Dogma 模型的图片""")
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(