From fe4eaf0bd1cff9a5fecb39b41b6cb8de9b0ca439 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Mon, 19 Feb 2024 15:00:41 -0800 Subject: [PATCH] Move the node into the package that gets scanned --- comfy_extras/{ => nodes}/nodes_stable_cascade.py | 6 +++--- tests/compare/__init__.py | 0 2 files changed, 3 insertions(+), 3 deletions(-) rename comfy_extras/{ => nodes}/nodes_stable_cascade.py (96%) create mode 100644 tests/compare/__init__.py diff --git a/comfy_extras/nodes_stable_cascade.py b/comfy_extras/nodes/nodes_stable_cascade.py similarity index 96% rename from comfy_extras/nodes_stable_cascade.py rename to comfy_extras/nodes/nodes_stable_cascade.py index b795d0083..517e27bfc 100644 --- a/comfy_extras/nodes_stable_cascade.py +++ b/comfy_extras/nodes/nodes_stable_cascade.py @@ -17,7 +17,7 @@ """ import torch -import nodes +import comfy.nodes.common import comfy.utils @@ -28,8 +28,8 @@ class StableCascade_EmptyLatentImage: @classmethod def INPUT_TYPES(s): return {"required": { - "width": ("INT", {"default": 1024, "min": 256, "max": nodes.MAX_RESOLUTION, "step": 8}), - "height": ("INT", {"default": 1024, "min": 256, "max": nodes.MAX_RESOLUTION, "step": 8}), + "width": ("INT", {"default": 1024, "min": 256, "max": comfy.nodes.common.MAX_RESOLUTION, "step": 8}), + "height": ("INT", {"default": 1024, "min": 256, "max": comfy.nodes.common.MAX_RESOLUTION, "step": 8}), "compression": ("INT", {"default": 42, "min": 4, "max": 128, "step": 1}), "batch_size": ("INT", {"default": 1, "min": 1, "max": 4096}) }} diff --git a/tests/compare/__init__.py b/tests/compare/__init__.py new file mode 100644 index 000000000..e69de29bb