Update models.py (#2)

- Update models.py (5b349aca014f8d204c87e7bdd4c9970bff5ff500)


Co-authored-by: John Smith <John6666@users.noreply.huggingface.co>

models.py

@@ -36,7 +36,7 @@ def load_and_compile_models():
     )
     # Load VAE separately as recommended for stabilityai models
     pipe_global.vae = AutoencoderKL.from_pretrained(
-        "stabilityai/sdxl-vae", torch_dtype=torch.float16, variant="fp16"
+        "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16,
     )
     pipe_global.to("cuda")
     print("SDXL base model loaded and moved to CUDA.")