Update app.py

app.py

@@ -1353,9 +1353,10 @@ def load_sdxl_base(device: str = "cuda"):
         torch_dtype=torch.float16
     ).to(device)
     
+    # Use fp16-fix VAE to avoid NaN issues with SDXL's original VAE in fp16
+    print("   Using madebyollin/sdxl-vae-fp16-fix for stable fp16 decoding...")
     vae = AutoencoderKL.from_pretrained(
-        "stabilityai/stable-diffusion-xl-base-1.0",
-        subfolder="vae",
+        "madebyollin/sdxl-vae-fp16-fix",
         torch_dtype=torch.float16
     ).to(device)
     
@@ -1772,7 +1773,7 @@ def create_demo():
                     
                     lyra_checkpoint = gr.Textbox(
                         label="Lyra VAE Checkpoint (optional)",
-                        value="weights/lyra_illustrious_step_41000.safetensors",
+                        value="",
                         placeholder="e.g., lyra_e100_s50000.safetensors",
                         info="Leave empty for latest. Loaded from weights/ folder in Lyra repo."
                     )