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"))