alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-25 15:15:45 -03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #202 from JaredTherriault/patch-1

Browse Source 
Resolve issue with loading vae
This commit is contained in:
TSC
2023-10-18 17:46:31 -05:00
committed by GitHub
parent 6921c58d8d c1c4c97469
commit 749c42b69b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tsc_utils.py
View File

@@ -250,7 +250,8 @@ def load_vae(vae_name, id, cache=None, cache_overwrite=False):
vae_path = vae_name vae_path = vae_name
else: else:
vae_path = folder_paths.get_full_path("vae", vae_name) 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 cache:
if len([entry for entry in loaded_objects["vae"] if id in entry[-1]]) < cache: if len([entry for entry in loaded_objects["vae"] if id in entry[-1]]) < cache: