xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 23:37:09 +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 # # For weights without clear prefix, assume they're decoder weights
# sd["taehv_decoder.{}".format(k)] = tae_weights[k] # sd["taehv_decoder.{}".format(k)] = tae_weights[k]
sa.update(tae_weights) sd.update(tae_weights)
sa['taehv_flag'] = True sd['taehv_flag'] = True
#TODO: Confirm scale/shift params #TODO: Confirm scale/shift params
if name == "taehv": if name == "taehv":
sd["vae_scale"] = torch.tensor(0.476986) # HunyuanVideo scale sd["vae_scale"] = torch.tensor(0.476986) # HunyuanVideo scale
@ -808,7 +808,7 @@ class VAELoader:
def load_vae(self, vae_name): def load_vae(self, vae_name):
if vae_name in ["taesd", "taesdxl", "taesd3", "taef1"]: if vae_name in ["taesd", "taesdxl", "taesd3", "taef1"]:
sd = self.load_taesd(vae_name) 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) sd = self.load_tae_video(vae_name)
else: else:
vae_path = folder_paths.get_full_path_or_raise("vae", vae_name) vae_path = folder_paths.get_full_path_or_raise("vae", vae_name)