From 87d6fd0c35cd34159639a22300c7ac8e7eef9886 Mon Sep 17 00:00:00 2001
From: lilly <lilly1987@naver.com>
Date: Wed, 25 Oct 2023 00:34:18 +0900
Subject: [PATCH] Update VAELoaderText.py

---
 VAELoaderText.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/VAELoaderText.py b/VAELoaderText.py
index d695378..144f23e 100644
--- a/VAELoaderText.py
+++ b/VAELoaderText.py
@@ -27,8 +27,11 @@ class VAELoaderText:
     def load_vae(self, vae_name):
         print(f"[{ccolor}]vae_name : [/{ccolor}]", vae_name)
         vae_path=getFullPath(vae_name,"vae")
+        
+        
         try:
-            vae = comfy.sd.VAE(ckpt_path=vae_path)
+            sd = comfy.utils.load_torch_file(vae_path)
+            vae = comfy.sd.VAE(sd=sd)
             return (vae,)
         except Exception as e:
             console.print_exception()