From bdbe3db2a972fa0d38153da045df66511fc293d8 Mon Sep 17 00:00:00 2001 From: shadowcz007 Date: Thu, 21 Dec 2023 10:39:32 +0800 Subject: [PATCH] Update Vae.py --- nodes/Vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes/Vae.py b/nodes/Vae.py index 413cbc9..8ef8d44 100644 --- a/nodes/Vae.py +++ b/nodes/Vae.py @@ -145,7 +145,7 @@ class VAELoader: RETURN_TYPES = ("VAE",) FUNCTION = "load_vae" - CATEGORY = "♾️Mixlab/ConsistencyDecoder" + CATEGORY = "♾️Mixlab/_test" #TODO: scale factor? def load_vae(self, vae_name): @@ -165,7 +165,7 @@ class VAEDecode: RETURN_TYPES = ("IMAGE",) FUNCTION = "decode" - CATEGORY = "♾️Mixlab/ConsistencyDecoder" + CATEGORY = "♾️Mixlab/_test" def decode(self, vae, samples): image = vae.decode(samples["samples"].to("cuda:0"))