From 16fe53f0a32b741db999971426c26813dfa9c0be Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Sun, 3 Dec 2023 20:27:40 +0530 Subject: [PATCH] Update base.py --- base.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/base.py b/base.py index 0986159..b00e64e 100644 --- a/base.py +++ b/base.py @@ -56,7 +56,7 @@ class SeedContext(): # import comfy.model_base as BaseModel - +# Node groups------------------------------ class xy_Tiling_KSampler: def __init__(self): pass @@ -86,7 +86,7 @@ class xy_Tiling_KSampler: RETURN_NAMES = ("latent", "progress_latent") FUNCTION = "sample" - CATEGORY = "Tiled/Sampling" + CATEGORY = "Jags_vector/xy_tile_sampler" def apply_asymmetric_tiling(self, model, tileX, tileY): for layer in [layer for layer in model.modules() if isinstance(layer, torch.nn.Conv2d)]: @@ -152,7 +152,7 @@ class CircularVAEDecode: RETURN_TYPES = ("IMAGE",) FUNCTION = "decode" - CATEGORY = "latent" + CATEGORY = "Jags_vector/latent" def decode(self, vae, samples): for layer in [layer for layer in vae.first_stage_model.modules() if isinstance(layer, torch.nn.Conv2d)]: @@ -161,7 +161,11 @@ class CircularVAEDecode: NODE_CLASS_MAPPINGS = { "xy_Tiling_KSampler": xy_Tiling_KSampler, - "CircularVAEDecode": CircularVAEDecode, + "CircularVAEDecode": CircularVAEDecode +} +NODE_DISPLAY_NAME_MAPPINGS = { + "xy_Tiling_KSampler": 'Jags-XY_tile sampler', + "CircularVAEDecode": 'Jags-CircularVAEDecode' }