From c1c4c974695051fb2c62791fccf66266946dae20 Mon Sep 17 00:00:00 2001
From: JaredTherriault <noirjt@live.com>
Date: Wed, 18 Oct 2023 11:59:20 -0700
Subject: [PATCH] Update tsc_utils.py

Resolve issue with loading vae
---
 tsc_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tsc_utils.py b/tsc_utils.py
index 87d8777..ca138d9 100644
--- a/tsc_utils.py
+++ b/tsc_utils.py
@@ -250,7 +250,8 @@ def load_vae(vae_name, id, cache=None, cache_overwrite=False):
         vae_path = vae_name
     else:
         vae_path = folder_paths.get_full_path("vae", vae_name)
-    vae = comfy.sd.VAE(ckpt_path=vae_path)
+    sd = comfy.utils.load_torch_file(vae_path)
+    vae = comfy.sd.VAE(sd=sd)
 
     if cache:
         if len([entry for entry in loaded_objects["vae"] if id in entry[-1]]) < cache: