fix delete vae error

2024-09-28 12:11:05 +02:00
parent 227b3ae926
commit c5e3c8cf14
1 changed files with 2 additions and 2 deletions
@@ -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,