Update Vae.py

2023-12-21 10:39:32 +08:00
parent 27ae99ad86
commit bdbe3db2a9
1 changed files with 2 additions and 2 deletions
@@ -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"))