Update VAELoaderText.py

2023-10-25 00:34:18 +09:00
parent 70ed2501bb
commit 87d6fd0c35
1 changed files with 4 additions and 1 deletions
@@ -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()