From 42232f4d20f05b6a1778989ff4f254f9f25233fb Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Wed, 3 Jan 2024 20:12:33 -0800 Subject: [PATCH] fix module ref --- comfy/nodes/base_nodes.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/comfy/nodes/base_nodes.py b/comfy/nodes/base_nodes.py index 068cd6c95..ae687ae39 100644 --- a/comfy/nodes/base_nodes.py +++ b/comfy/nodes/base_nodes.py @@ -600,11 +600,11 @@ class VAELoader: encoder = next(filter(lambda a: a.startswith("{}_encoder.".format(name)), approx_vaes)) decoder = next(filter(lambda a: a.startswith("{}_decoder.".format(name)), approx_vaes)) - enc = comfy.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: sd["taesd_encoder.{}".format(k)] = enc[k] - dec = comfy.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: sd["taesd_decoder.{}".format(k)] = dec[k] @@ -931,7 +931,7 @@ class GLIGENTextBoxApply: class EmptyLatentImage: def __init__(self): - self.device = comfy.model_management.intermediate_device() + self.device = model_management.intermediate_device() @classmethod def INPUT_TYPES(s): @@ -1773,4 +1773,4 @@ NODE_DISPLAY_NAME_MAPPINGS = { # _for_testing "VAEDecodeTiled": "VAE Decode (Tiled)", "VAEEncodeTiled": "VAE Encode (Tiled)", -} \ No newline at end of file +}