wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-24 21:30:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix #5 TAESD node was using a bad variable name that shadowed a module in a relative import

Browse Source
This commit is contained in:
doctorpangloss 2024-03-29 16:28:13 -07:00
parent bd87697fdf
commit b0ab12bf05
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
comfy/nodes/base_nodes.py
View File

@ -666,7 +666,7 @@ class VAELoader:
@staticmethod @staticmethod
def load_taesd(name): def load_taesd(name):
sd = {} sd_ = {}
approx_vaes = folder_paths.get_filename_list("vae_approx") approx_vaes = folder_paths.get_filename_list("vae_approx")
encoder = next(filter(lambda a: a.startswith("{}_encoder.".format(name)), approx_vaes)) encoder = next(filter(lambda a: a.startswith("{}_encoder.".format(name)), approx_vaes))
@ -674,17 +674,17 @@ class VAELoader:
enc = utils.load_torch_file(folder_paths.get_full_path("vae_approx", encoder)) enc = utils.load_torch_file(folder_paths.get_full_path("vae_approx", encoder))
for k in enc: for k in enc:
sd["taesd_encoder.{}".format(k)] = enc[k] sd_["taesd_encoder.{}".format(k)] = enc[k]
dec = utils.load_torch_file(folder_paths.get_full_path("vae_approx", decoder)) dec = utils.load_torch_file(folder_paths.get_full_path("vae_approx", decoder))
for k in dec: for k in dec:
sd["taesd_decoder.{}".format(k)] = dec[k] sd_["taesd_decoder.{}".format(k)] = dec[k]
if name == "taesd": if name == "taesd":
sd["vae_scale"] = torch.tensor(0.18215) sd_["vae_scale"] = torch.tensor(0.18215)
elif name == "taesdxl": elif name == "taesdxl":
sd["vae_scale"] = torch.tensor(0.13025) sd_["vae_scale"] = torch.tensor(0.13025)
return sd return sd_
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -697,11 +697,11 @@ class VAELoader:
#TODO: scale factor? #TODO: scale factor?
def load_vae(self, vae_name): def load_vae(self, vae_name):
if vae_name in ["taesd", "taesdxl"]: if vae_name in ["taesd", "taesdxl"]:
sd = self.load_taesd(vae_name) sd_ = self.load_taesd(vae_name)
else: else:
vae_path = folder_paths.get_full_path("vae", vae_name) vae_path = folder_paths.get_full_path("vae", vae_name)
sd = utils.load_torch_file(vae_path) sd_ = utils.load_torch_file(vae_path)
vae = sd.VAE(sd=sd) vae = sd.VAE(sd=sd_)
return (vae,) return (vae,)
class ControlNetLoader: class ControlNetLoader: