wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 06:10:50 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

fix linting errors

Browse Source
This commit is contained in:
doctorpangloss 2025-07-16 15:52:36 -07:00
parent 94101b5b86
commit dd3925ec3b
6 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy_extras/nodes/nodes_cosmos_upsampling.py
View File

@ -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: "

2
comfy_extras/nodes/nodes_flux.py
View File

@ -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:

4
comfy_extras/nodes/nodes_ideogram.py
View File

@ -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 ---

2
comfy_extras/nodes/nodes_images.py
View File

@ -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:

2
comfy_extras/nodes/nodes_inpainting.py
View File

@ -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):

2
comfy_extras/nodes/nodes_wan_prompt_expander.py
View File

@ -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",