From c5e3c8cf145b1044ad46a76b05cc278337751990 Mon Sep 17 00:00:00 2001
From: GiusTex <112352961+GiusTex@users.noreply.github.com>
Date: Sat, 28 Sep 2024 12:11:05 +0200
Subject: [PATCH] fix delete vae error

---
 nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index eb66dc3..c828a50 100644
--- a/nodes.py
+++ b/nodes.py
@@ -211,7 +211,7 @@ class LoadDiffusersOutpaintModels:
         pipe = StableDiffusionXLFillPipeline.from_pretrained(
             f"{model_path}",
             torch_dtype=torch.float16,
-            vae=vae_model,
+            vae=vae,
             controlnet=controlnet_model,
             variant="fp16",
         ).to("cuda")
@@ -220,7 +220,7 @@ class LoadDiffusersOutpaintModels:
         
         diffusers_outpaint_pipe = {
             "pipe": pipe,
-            "vae": vae,
+            "vae": vae_model,
             "model": model,
             "controlnet_model": controlnet_model,
             "state_dict": state_dict,