Commit 55f37baa authored by comfyanonymous's avatar comfyanonymous
Browse files

Move some stable cascade nodes outside of _for_testing.

parent 3f75419e
...@@ -37,7 +37,7 @@ class StableCascade_EmptyLatentImage: ...@@ -37,7 +37,7 @@ class StableCascade_EmptyLatentImage:
RETURN_NAMES = ("stage_c", "stage_b") RETURN_NAMES = ("stage_c", "stage_b")
FUNCTION = "generate" FUNCTION = "generate"
CATEGORY = "_for_testing/stable_cascade" CATEGORY = "latent/stable_cascade"
def generate(self, width, height, compression, batch_size=1): def generate(self, width, height, compression, batch_size=1):
c_latent = torch.zeros([batch_size, 16, height // compression, width // compression]) c_latent = torch.zeros([batch_size, 16, height // compression, width // compression])
...@@ -63,7 +63,7 @@ class StableCascade_StageC_VAEEncode: ...@@ -63,7 +63,7 @@ class StableCascade_StageC_VAEEncode:
RETURN_NAMES = ("stage_c", "stage_b") RETURN_NAMES = ("stage_c", "stage_b")
FUNCTION = "generate" FUNCTION = "generate"
CATEGORY = "_for_testing/stable_cascade" CATEGORY = "latent/stable_cascade"
def generate(self, image, vae, compression): def generate(self, image, vae, compression):
width = image.shape[-2] width = image.shape[-2]
...@@ -91,7 +91,7 @@ class StableCascade_StageB_Conditioning: ...@@ -91,7 +91,7 @@ class StableCascade_StageB_Conditioning:
FUNCTION = "set_prior" FUNCTION = "set_prior"
CATEGORY = "_for_testing/stable_cascade" CATEGORY = "conditioning/stable_cascade"
def set_prior(self, conditioning, stage_c): def set_prior(self, conditioning, stage_c):
c = [] c = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment