diff --git a/comfy_extras/nodes/nodes_cosmos_upsampling.py b/comfy_extras/nodes/nodes_cosmos_upsampling.py index 0e8912732..b709457b4 100644 --- a/comfy_extras/nodes/nodes_cosmos_upsampling.py +++ b/comfy_extras/nodes/nodes_cosmos_upsampling.py @@ -15,8 +15,7 @@ from comfy.model_management import load_models_gpu from comfy.model_patcher import ModelPatcher from comfy.node_helpers import export_custom_nodes from comfy.nodes.package_typing import InputTypes, ValidatedNodeResult, CustomNode -from comfy_extras.nodes.nodes_language import TransformersLoader, TransformersTokenize, OneShotInstructTokenize, \ - _AUTO_CHAT_TEMPLATE +from .nodes_language import TransformersLoader, TransformersTokenize, OneShotInstructTokenize, _AUTO_CHAT_TEMPLATE # from https://github.com/NVIDIA/Cosmos/blob/b867572b99d08f450ddb8bcd6661d8c35bf6b967/cosmos1/models/diffusion/nemo/inference/inference_utils.py#L54 COSMOS_TEXT_TO_WORLD_UPSAMPLE_TASK = "Upsample the short caption to a long caption: " diff --git a/comfy_extras/nodes/nodes_flux.py b/comfy_extras/nodes/nodes_flux.py index 6e6e51f4d..16a7007e0 100644 --- a/comfy_extras/nodes/nodes_flux.py +++ b/comfy_extras/nodes/nodes_flux.py @@ -1,6 +1,6 @@ from comfy import node_helpers import comfy.utils -from comfy_extras.constants.resolutions import KONTEXT_RESOLUTIONS +from ..constants.resolutions import KONTEXT_RESOLUTIONS class CLIPTextEncodeFlux: diff --git a/comfy_extras/nodes/nodes_ideogram.py b/comfy_extras/nodes/nodes_ideogram.py index 251d9ccfe..842c033a1 100644 --- a/comfy_extras/nodes/nodes_ideogram.py +++ b/comfy_extras/nodes/nodes_ideogram.py @@ -11,8 +11,8 @@ from PIL import Image from comfy.component_model.tensor_types import RGBImageBatch, MaskBatch, ImageBatch from comfy.nodes.package_typing import CustomNode, Seed31 from comfy.utils import pil2tensor, tensor2pil -from comfy_extras.constants.resolutions import IDEOGRAM_RESOLUTIONS -from comfy_extras.nodes.nodes_mask import MaskToImage +from ..constants.resolutions import IDEOGRAM_RESOLUTIONS +from .nodes_mask import MaskToImage # --- ENUMs and Constants --- diff --git a/comfy_extras/nodes/nodes_images.py b/comfy_extras/nodes/nodes_images.py index 249cef10e..bdd4ef4f7 100644 --- a/comfy_extras/nodes/nodes_images.py +++ b/comfy_extras/nodes/nodes_images.py @@ -19,7 +19,7 @@ from comfy.nodes.base_nodes import ImageScale from comfy.nodes.common import MAX_RESOLUTION from comfy.nodes.package_typing import CustomNode from comfy.utils import common_upscale -from comfy_extras.constants.resolutions import RESOLUTION_MAP, RESOLUTION_NAMES, SD_RESOLUTIONS +from ..constants.resolutions import RESOLUTION_MAP, RESOLUTION_NAMES, SD_RESOLUTIONS def levels_adjustment(image: ImageBatch, black_level: float = 0.0, mid_level: float = 0.5, white_level: float = 1.0, clip: bool = True) -> ImageBatch: diff --git a/comfy_extras/nodes/nodes_inpainting.py b/comfy_extras/nodes/nodes_inpainting.py index aa7399ba9..d1e1abe77 100644 --- a/comfy_extras/nodes/nodes_inpainting.py +++ b/comfy_extras/nodes/nodes_inpainting.py @@ -7,7 +7,7 @@ from torch import Tensor from comfy.component_model.tensor_types import MaskBatch, ImageBatch from comfy.nodes.package_typing import CustomNode -from comfy_extras.constants.resolutions import RESOLUTION_MAP, SD_RESOLUTIONS, RESOLUTION_NAMES +from ..constants.resolutions import RESOLUTION_MAP, SD_RESOLUTIONS, RESOLUTION_NAMES class CompositeContext(NamedTuple): diff --git a/comfy_extras/nodes/nodes_wan_prompt_expander.py b/comfy_extras/nodes/nodes_wan_prompt_expander.py index 910bb2491..6ed5fd3d4 100644 --- a/comfy_extras/nodes/nodes_wan_prompt_expander.py +++ b/comfy_extras/nodes/nodes_wan_prompt_expander.py @@ -4,7 +4,7 @@ import torch from comfy.language.language_types import LanguageModel from comfy.node_helpers import export_custom_nodes from comfy.nodes.package_typing import InputTypes, ValidatedNodeResult -from comfy_extras.nodes.nodes_language import OneShotInstructTokenize, TransformersLoader +from .nodes_language import OneShotInstructTokenize, TransformersLoader model_dict = { "QwenVL2.5_3B": "Qwen/Qwen2.5-VL-3B-Instruct",