Skip to content

Commit ff6ca2a

Browse files
Move PAG to model_patches/unet section.
Move other unet model_patches nodes to model_patches/unet section.
1 parent 374e093 commit ff6ca2a

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

comfy_extras/nodes_freelunch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def INPUT_TYPES(s):
3434
RETURN_TYPES = ("MODEL",)
3535
FUNCTION = "patch"
3636

37-
CATEGORY = "model_patches"
37+
CATEGORY = "model_patches/unet"
3838

3939
def patch(self, model, b1, b2, s1, s2):
4040
model_channels = model.model.model_config.unet_config["model_channels"]
@@ -73,7 +73,7 @@ def INPUT_TYPES(s):
7373
RETURN_TYPES = ("MODEL",)
7474
FUNCTION = "patch"
7575

76-
CATEGORY = "model_patches"
76+
CATEGORY = "model_patches/unet"
7777

7878
def patch(self, model, b1, b2, s1, s2):
7979
model_channels = model.model.model_config.unet_config["model_channels"]

comfy_extras/nodes_hypertile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def INPUT_TYPES(s):
3232
RETURN_TYPES = ("MODEL",)
3333
FUNCTION = "patch"
3434

35-
CATEGORY = "model_patches"
35+
CATEGORY = "model_patches/unet"
3636

3737
def patch(self, model, tile_size, swap_size, max_depth, scale_depth):
3838
model_channels = model.model.model_config.unet_config["model_channels"]

comfy_extras/nodes_pag.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def INPUT_TYPES(s):
1919
RETURN_TYPES = ("MODEL",)
2020
FUNCTION = "patch"
2121

22-
CATEGORY = "_for_testing"
22+
CATEGORY = "model_patches/unet"
2323

2424
def patch(self, model, scale):
2525
unet_block = "middle"

0 commit comments

Comments
 (0)