xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 19:47:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed syntax errors

Browse Source
This commit is contained in:
Pratik-Doshi-99 2025-07-17 01:25:16 +00:00
parent 6dde0bab3f
commit 596c342668
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -785,8 +785,8 @@ class VAELoader:
# # For weights without clear prefix, assume they're decoder weights
# sd["taehv_decoder.{}".format(k)] = tae_weights[k]
sa.update(tae_weights)
sa['taehv_flag'] = True
sd.update(tae_weights)
sd['taehv_flag'] = True
#TODO: Confirm scale/shift params
if name == "taehv":
sd["vae_scale"] = torch.tensor(0.476986) # HunyuanVideo scale
@ -808,7 +808,7 @@ class VAELoader:
def load_vae(self, vae_name):
if vae_name in ["taesd", "taesdxl", "taesd3", "taef1"]:
sd = self.load_taesd(vae_name)
elif vae_name in ["taehv", "taew2_1"]
elif vae_name in ["taehv", "taew2_1"]:
sd = self.load_tae_video(vae_name)
else:
vae_path = folder_paths.get_full_path_or_raise("vae", vae_name)