Add I2V for causal forcing model.

.github/workflows/openapi-lint.yml

@@ -1,31 +0,0 @@
-name: OpenAPI Lint
-
-on:
-  pull_request:
-    paths:
-      - 'openapi.yaml'
-      - '.spectral.yaml'
-      - '.github/workflows/openapi-lint.yml'
-
-permissions:
-  contents: read
-
-jobs:
-  spectral:
-    name: Run Spectral
-    runs-on: ubuntu-latest
-
-    steps:
-      - name: Checkout repository
-        uses: actions/checkout@v4
-
-      - name: Set up Node.js
-        uses: actions/setup-node@v4
-        with:
-          node-version: '20'
-
-      - name: Install Spectral
-        run: npm install -g @stoplight/spectral-cli@6
-
-      - name: Lint openapi.yaml
-        run: spectral lint openapi.yaml --ruleset .spectral.yaml --fail-severity=error

.gitignore

@@ -23,4 +23,3 @@ web_custom_versions/
 .DS_Store
 filtered-openapi.yaml
 uv.lock
-.comfy_environment

.spectral.yaml

@@ -1,91 +0,0 @@
-extends:
-  - spectral:oas
-
-# Severity levels: error, warn, info, hint, off
-# Rules from the built-in "spectral:oas" ruleset are active by default.
-# Below we tune severity and add custom rules for our conventions.
-#
-# This ruleset mirrors Comfy-Org/cloud/.spectral.yaml so specs across the
-# organization are linted against a single consistent standard.
-
-rules:
-  # -----------------------------------------------------------------------
-  # Built-in rule severity overrides
-  # -----------------------------------------------------------------------
-  operation-operationId: error
-  operation-description: warn
-  operation-tag-defined: error
-  info-contact: off
-  info-description: warn
-  no-eval-in-markdown: error
-  no-$ref-siblings: error
-
-  # -----------------------------------------------------------------------
-  # Custom rules: naming conventions
-  # -----------------------------------------------------------------------
-
-  # Property names should be snake_case
-  property-name-snake-case:
-    description: Property names must be snake_case
-    severity: warn
-    given: "$.components.schemas.*.properties[*]~"
-    then:
-      function: pattern
-      functionOptions:
-        match: "^[a-z][a-z0-9]*(_[a-z0-9]+)*$"
-
-  # Operation IDs should be camelCase
-  operation-id-camel-case:
-    description: Operation IDs must be camelCase
-    severity: warn
-    given: "$.paths.*.*.operationId"
-    then:
-      function: pattern
-      functionOptions:
-        match: "^[a-z][a-zA-Z0-9]*$"
-
-  # -----------------------------------------------------------------------
-  # Custom rules: response conventions
-  # -----------------------------------------------------------------------
-
-  # Error responses (4xx, 5xx) should use a consistent shape
-  error-response-schema:
-    description: Error responses should reference a standard error schema
-    severity: hint
-    given: "$.paths.*.*.responses[?(@property >= '400' && @property < '600')].content['application/json'].schema"
-    then:
-      field: "$ref"
-      function: truthy
-
-  # All 2xx responses with JSON body should have a schema
-  response-schema-defined:
-    description: Success responses with JSON content should define a schema
-    severity: warn
-    given: "$.paths.*.*.responses[?(@property >= '200' && @property < '300')].content['application/json']"
-    then:
-      field: schema
-      function: truthy
-
-  # -----------------------------------------------------------------------
-  # Custom rules: best practices
-  # -----------------------------------------------------------------------
-
-  # Path parameters must have a description
-  path-param-description:
-    description: Path parameters should have a description
-    severity: warn
-    given:
-      - "$.paths.*.parameters[?(@.in == 'path')]"
-      - "$.paths.*.*.parameters[?(@.in == 'path')]"
-    then:
-      field: description
-      function: truthy
-
-  # Schemas should have a description
-  schema-description:
-    description: Component schemas should have a description
-    severity: hint
-    given: "$.components.schemas.*"
-    then:
-      field: description
-      function: truthy

README.md

@@ -133,7 +133,7 @@ Workflow examples can be found on the [Examples page](https://comfyanonymous.git
 ComfyUI follows a weekly release cycle targeting Monday but this regularly changes because of model releases or large changes to the codebase. There are three interconnected repositories:
 
 1. **[ComfyUI Core](https://github.com/comfyanonymous/ComfyUI)**
-   - Releases a new major stable version (e.g., v0.7.0) roughly every 2 weeks.
+   - Releases a new stable version (e.g., v0.7.0) roughly every week.
    - Starting from v0.4.0 patch versions will be used for fixes backported onto the current stable release.
    - Minor versions will be used for releases off the master branch.
    - Patch versions may still be used for releases on the master branch in cases where a backport would not make sense.

comfy/cli_args.py

@@ -238,8 +238,6 @@ database_default_path = os.path.abspath(
 )
 parser.add_argument("--database-url", type=str, default=f"sqlite:///{database_default_path}", help="Specify the database URL, e.g. for an in-memory database you can use 'sqlite:///:memory:'.")
 parser.add_argument("--enable-assets", action="store_true", help="Enable the assets system (API routes, database synchronization, and background scanning).")
-parser.add_argument("--feature-flag", type=str, action='append', default=[], metavar="KEY[=VALUE]", help="Set a server feature flag. Use KEY=VALUE to set an explicit value, or bare KEY to set it to true. Can be specified multiple times. Boolean values (true/false) and numbers are auto-converted. Examples: --feature-flag show_signin_button=true  or  --feature-flag show_signin_button")
-parser.add_argument("--list-feature-flags", action="store_true", help="Print the registry of known CLI-settable feature flags as JSON and exit.")
 
 if comfy.options.args_parsing:
     args = parser.parse_args()

comfy/deploy_environment.py

@@ -1,34 +0,0 @@
-import functools
-import logging
-import os
-
-logger = logging.getLogger(__name__)
-
-_DEFAULT_DEPLOY_ENV = "local-git"
-_ENV_FILENAME = ".comfy_environment"
-
-# Resolve the ComfyUI install directory (the parent of this `comfy/` package).
-# We deliberately avoid `folder_paths.base_path` here because that is overridden
-# by the `--base-directory` CLI arg to a user-supplied path, whereas the
-# `.comfy_environment` marker is written by launchers/installers next to the
-# ComfyUI install itself.
-_COMFY_INSTALL_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
-
-
-@functools.cache
-def get_deploy_environment() -> str:
-    env_file = os.path.join(_COMFY_INSTALL_DIR, _ENV_FILENAME)
-    try:
-        with open(env_file, encoding="utf-8") as f:
-            # Cap the read so a malformed or maliciously crafted file (e.g.
-            # a single huge line with no newline) can't blow up memory.
-            first_line = f.readline(128).strip()
-            value = "".join(c for c in first_line if 32 <= ord(c) < 127)
-            if value:
-                return value
-    except FileNotFoundError:
-        pass
-    except Exception as e:
-        logger.error("Failed to read %s: %s", env_file, e)
-
-    return _DEFAULT_DEPLOY_ENV

comfy/k_diffusion/sampling.py

@@ -1859,6 +1859,23 @@ def sample_ar_video(model, x, sigmas, extra_args=None, callback=None, disable=No
     output = torch.zeros_like(x)
     s_in = x.new_ones([x.shape[0]])
     current_start_frame = 0
+
+    # I2V: seed KV cache with the initial image latent before the denoising loop
+    initial_latent = ar_config.get("initial_latent", None)
+    if initial_latent is not None:
+        initial_latent = inner_model.process_latent_in(initial_latent).to(device=device, dtype=model_dtype)
+        n_init = initial_latent.shape[2]
+        output[:, :, :n_init] = initial_latent
+
+        ar_state = {"start_frame": 0, "kv_caches": kv_caches, "crossattn_caches": crossattn_caches}
+        transformer_options["ar_state"] = ar_state
+        zero_sigma = sigmas.new_zeros([1])
+        _ = model(initial_latent, zero_sigma * s_in, **extra_args)
+
+        current_start_frame = n_init
+        remaining = lat_t - n_init
+        num_blocks = -(-remaining // num_frame_per_block)
+
     num_sigma_steps = len(sigmas) - 1
     total_real_steps = num_blocks * num_sigma_steps
     step_count = 0

comfy/model_management.py

@@ -721,15 +721,13 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu
     else:
         minimum_memory_required = max(inference_memory, minimum_memory_required + extra_reserved_memory())
 
-    # Order-preserving dedup. A plain set() would randomize iteration order across runs
-    models_temp = {}
+    models_temp = set()
     for m in models:
-        models_temp[m] = None
+        models_temp.add(m)
         for mm in m.model_patches_models():
-            models_temp[mm] = None
+            models_temp.add(mm)
 
-    models = list(models_temp)
-    models.reverse()
+    models = models_temp
 
     models_to_load = []
 

comfy/model_prefetch.py

@@ -37,8 +37,7 @@ def prefetch_queue_pop(queue, device, module):
     consumed = queue.pop(0)
     if consumed is not None:
         offload_stream, prefetch_state = consumed
-        if offload_stream is not None:
-            offload_stream.wait_stream(comfy.model_management.current_stream(device))
+        offload_stream.wait_stream(comfy.model_management.current_stream(device))
         _, comfy_modules = prefetch_state
         if comfy_modules is not None:
             cleanup_prefetched_modules(comfy_modules)

comfy/ops.py

@@ -253,9 +253,6 @@ def resolve_cast_module_with_vbar(s, dtype, device, bias_dtype, compute_dtype, w
     if bias is not None:
         bias = post_cast(s, "bias", bias, bias_dtype, prefetch["resident"], update_weight)
 
-    if prefetch["signature"] is not None:
-        prefetch["resident"] = True
-
     return weight, bias
 
 

comfy/sampler_helpers.py

@@ -89,8 +89,7 @@ def get_additional_models(conds, dtype):
         gligen += get_models_from_cond(conds[k], "gligen")
         add_models += get_models_from_cond(conds[k], "additional_models")
 
-    # Order-preserving dedup. A plain set() would randomize iteration order across runs
-    control_nets = list(dict.fromkeys(cnets))
+    control_nets = set(cnets)
 
     inference_memory = 0
     control_models = []

comfy_api/feature_flags.py

@@ -5,95 +5,12 @@ This module handles capability negotiation between frontend and backend,
 allowing graceful protocol evolution while maintaining backward compatibility.
 """
 
-import logging
-from typing import Any, TypedDict
+from typing import Any
 
 from comfy.cli_args import args
 
-
-class FeatureFlagInfo(TypedDict):
-    type: str
-    default: Any
-    description: str
-
-
-# Registry of known CLI-settable feature flags.
-# Launchers can query this via --list-feature-flags to discover valid flags.
-CLI_FEATURE_FLAG_REGISTRY: dict[str, FeatureFlagInfo] = {
-    "show_signin_button": {
-        "type": "bool",
-        "default": False,
-        "description": "Show the sign-in button in the frontend even when not signed in",
-    },
-}
-
-
-def _coerce_bool(v: str) -> bool:
-    """Strict bool coercion: only 'true'/'false' (case-insensitive).
-
-    Anything else raises ValueError so the caller can warn and drop the flag,
-    rather than silently treating typos like 'ture' or 'yes' as False.
-    """
-    lower = v.lower()
-    if lower == "true":
-        return True
-    if lower == "false":
-        return False
-    raise ValueError(f"expected 'true' or 'false', got {v!r}")
-
-
-_COERCE_FNS: dict[str, Any] = {
-    "bool": _coerce_bool,
-    "int": lambda v: int(v),
-    "float": lambda v: float(v),
-}
-
-
-def _coerce_flag_value(key: str, raw_value: str) -> Any:
-    """Coerce a raw string value using the registry type, or keep as string.
-
-    Returns the raw string if the key is unregistered or the type is unknown.
-    Raises ValueError/TypeError if the key is registered with a known type but
-    the value cannot be coerced; callers are expected to warn and drop the flag.
-    """
-    info = CLI_FEATURE_FLAG_REGISTRY.get(key)
-    if info is None:
-        return raw_value
-    coerce = _COERCE_FNS.get(info["type"])
-    if coerce is None:
-        return raw_value
-    return coerce(raw_value)
-
-
-def _parse_cli_feature_flags() -> dict[str, Any]:
-    """Parse --feature-flag key=value pairs from CLI args into a dict.
-
-    Items without '=' default to the value 'true' (bare flag form).
-    Flags whose value cannot be coerced to the registered type are dropped
-    with a warning, so a typo like '--feature-flag some_bool=ture' does not
-    silently take effect as the wrong value.
-    """
-    result: dict[str, Any] = {}
-    for item in getattr(args, "feature_flag", []):
-        key, sep, raw_value = item.partition("=")
-        key = key.strip()
-        if not key:
-            continue
-        if not sep:
-            raw_value = "true"
-        try:
-            result[key] = _coerce_flag_value(key, raw_value.strip())
-        except (ValueError, TypeError) as e:
-            info = CLI_FEATURE_FLAG_REGISTRY.get(key, {})
-            logging.warning(
-                "Could not coerce --feature-flag %s=%r to %s (%s); dropping flag.",
-                key, raw_value.strip(), info.get("type", "?"), e,
-            )
-    return result
-
-
 # Default server capabilities
-_CORE_FEATURE_FLAGS: dict[str, Any] = {
+SERVER_FEATURE_FLAGS: dict[str, Any] = {
     "supports_preview_metadata": True,
     "max_upload_size": args.max_upload_size * 1024 * 1024, # Convert MB to bytes
     "extension": {"manager": {"supports_v4": True}},
@@ -101,11 +18,6 @@ _CORE_FEATURE_FLAGS: dict[str, Any] = {
     "assets": args.enable_assets,
 }
 
-# CLI-provided flags cannot overwrite core flags
-_cli_flags = {k: v for k, v in _parse_cli_feature_flags().items() if k not in _CORE_FEATURE_FLAGS}
-
-SERVER_FEATURE_FLAGS: dict[str, Any] = {**_CORE_FEATURE_FLAGS, **_cli_flags}
-
 
 def get_connection_feature(
     sockets_metadata: dict[str, dict[str, Any]],

comfy_api/latest/_io.py

@@ -395,6 +395,7 @@ class Combo(ComfyTypeIO):
 @comfytype(io_type="COMBO")
 class MultiCombo(ComfyTypeI):
     '''Multiselect Combo input (dropdown for selecting potentially more than one value).'''
+    # TODO: something is wrong with the serialization, frontend does not recognize it as multiselect
     Type = list[str]
     class Input(Combo.Input):
         def __init__(self, id: str, options: list[str], display_name: str=None, optional=False, tooltip: str=None, lazy: bool=None,
@@ -407,14 +408,12 @@ class MultiCombo(ComfyTypeI):
             self.default: list[str]
 
         def as_dict(self):
-            # Frontend expects `multi_select` to be an object config (not a boolean).
-            # Keep top-level `multiselect` from Combo.Input for backwards compatibility.
-            return super().as_dict() | prune_dict({
-                "multi_select": prune_dict({
-                    "placeholder": self.placeholder,
-                    "chip": self.chip,
-                }),
+            to_return = super().as_dict() | prune_dict({
+                "multi_select": self.multiselect,
+                "placeholder": self.placeholder,
+                "chip": self.chip,
             })
+            return to_return
 
 @comfytype(io_type="IMAGE")
 class Image(ComfyTypeIO):

comfy_api_nodes/apis/luma.py

@@ -1,12 +1,15 @@
 from __future__ import annotations
 
+
+import torch
+
 from enum import Enum
 from typing import Optional, Union
 
-import torch
 from pydantic import BaseModel, Field, confloat
 
 
+
 class LumaIO:
     LUMA_REF = "LUMA_REF"
     LUMA_CONCEPTS = "LUMA_CONCEPTS"
@@ -180,13 +183,13 @@ class LumaAssets(BaseModel):
 
 
 class LumaImageRef(BaseModel):
-    """Used for image gen"""
+    '''Used for image gen'''
     url: str = Field(..., description='The URL of the image reference')
     weight: confloat(ge=0.0, le=1.0) = Field(..., description='The weight of the image reference')
 
 
 class LumaImageReference(BaseModel):
-    """Used for video gen"""
+    '''Used for video gen'''
     type: Optional[str] = Field('image', description='Input type, defaults to image')
     url: str = Field(..., description='The URL of the image')
 
@@ -248,32 +251,3 @@ class LumaGeneration(BaseModel):
     assets: Optional[LumaAssets] = Field(None, description='The assets of the generation')
     model: str = Field(..., description='The model used for the generation')
     request: Union[LumaGenerationRequest, LumaImageGenerationRequest] = Field(..., description="The request used for the generation")
-
-
-class Luma2ImageRef(BaseModel):
-    url: str | None = None
-    data: str | None = None
-    media_type: str | None = None
-
-
-class Luma2GenerationRequest(BaseModel):
-    prompt: str = Field(..., min_length=1, max_length=6000)
-    model: str | None = None
-    type: str | None = None
-    aspect_ratio: str | None = None
-    style: str | None = None
-    output_format: str | None = None
-    web_search: bool | None = None
-    image_ref: list[Luma2ImageRef] | None = None
-    source: Luma2ImageRef | None = None
-
-
-class Luma2Generation(BaseModel):
-    id: str | None = None
-    type: str | None = None
-    state: str | None = None
-    model: str | None = None
-    created_at: str | None = None
-    output: list[LumaImageReference] | None = None
-    failure_reason: str | None = None
-    failure_code: str | None = None

comfy_api_nodes/apis/openai.py

@@ -56,14 +56,14 @@ class ModelResponseProperties(BaseModel):
     instructions: str | None = Field(None)
     max_output_tokens: int | None = Field(None)
     model: str | None = Field(None)
-    temperature: float | None = Field(None, description="Controls randomness in the response", ge=0.0, le=2.0)
+    temperature: float | None = Field(1, description="Controls randomness in the response", ge=0.0, le=2.0)
     top_p: float | None = Field(
-        None,
+        1,
         description="Controls diversity of the response via nucleus sampling",
         ge=0.0,
         le=1.0,
     )
-    truncation: str | None = Field(None, description="Allowed values: 'auto' or 'disabled'")
+    truncation: str | None = Field("disabled", description="Allowed values: 'auto' or 'disabled'")
 
 
 class ResponseProperties(BaseModel):

comfy_api_nodes/nodes_luma.py

@@ -1,11 +1,10 @@
+from typing import Optional
+
 import torch
 from typing_extensions import override
 
-from comfy_api.latest import IO, ComfyExtension, Input
+from comfy_api.latest import IO, ComfyExtension
 from comfy_api_nodes.apis.luma import (
-    Luma2Generation,
-    Luma2GenerationRequest,
-    Luma2ImageRef,
     LumaAspectRatio,
     LumaCharacterRef,
     LumaConceptChain,
@@ -31,7 +30,6 @@ from comfy_api_nodes.util import (
     download_url_to_video_output,
     poll_op,
     sync_op,
-    upload_image_to_comfyapi,
     upload_images_to_comfyapi,
     validate_string,
 )
@@ -214,9 +212,9 @@ class LumaImageGenerationNode(IO.ComfyNode):
         aspect_ratio: str,
         seed,
         style_image_weight: float,
-        image_luma_ref: LumaReferenceChain | None = None,
-        style_image: torch.Tensor | None = None,
-        character_image: torch.Tensor | None = None,
+        image_luma_ref: Optional[LumaReferenceChain] = None,
+        style_image: Optional[torch.Tensor] = None,
+        character_image: Optional[torch.Tensor] = None,
     ) -> IO.NodeOutput:
         validate_string(prompt, strip_whitespace=True, min_length=3)
         # handle image_luma_ref
@@ -436,7 +434,7 @@ class LumaTextToVideoGenerationNode(IO.ComfyNode):
         duration: str,
         loop: bool,
         seed,
-        luma_concepts: LumaConceptChain | None = None,
+        luma_concepts: Optional[LumaConceptChain] = None,
     ) -> IO.NodeOutput:
         validate_string(prompt, strip_whitespace=False, min_length=3)
         duration = duration if model != LumaVideoModel.ray_1_6 else None
@@ -535,6 +533,7 @@ class LumaImageToVideoGenerationNode(IO.ComfyNode):
             ],
             is_api_node=True,
             price_badge=PRICE_BADGE_VIDEO,
+
         )
 
     @classmethod
@@ -645,293 +644,6 @@ PRICE_BADGE_VIDEO = IO.PriceBadge(
 )
 
 
-def _luma2_uni1_common_inputs(max_image_refs: int) -> list:
-    return [
-        IO.Combo.Input(
-            "style",
-            options=["auto", "manga"],
-            default="auto",
-            tooltip="Style preset. 'auto' picks based on the prompt; "
-            "'manga' applies a manga/anime aesthetic and requires a portrait "
-            "aspect ratio (2:3, 9:16, 1:2, 1:3).",
-        ),
-        IO.Boolean.Input(
-            "web_search",
-            default=False,
-            tooltip="Search the web for visual references before generating.",
-        ),
-        IO.Autogrow.Input(
-            "image_ref",
-            template=IO.Autogrow.TemplateNames(
-                IO.Image.Input("image"),
-                names=[f"image_{i}" for i in range(1, max_image_refs + 1)],
-                min=0,
-            ),
-            optional=True,
-            tooltip=f"Up to {max_image_refs} reference images for style/content guidance.",
-        ),
-    ]
-
-
-async def _luma2_upload_image_refs(
-    cls: type[IO.ComfyNode],
-    refs: dict | None,
-    max_count: int,
-) -> list[Luma2ImageRef] | None:
-    if not refs:
-        return None
-    out: list[Luma2ImageRef] = []
-    for key in refs:
-        url = await upload_image_to_comfyapi(cls, refs[key])
-        out.append(Luma2ImageRef(url=url))
-    if len(out) > max_count:
-        raise ValueError(f"Maximum {max_count} reference images are allowed.")
-    return out or None
-
-
-async def _luma2_submit_and_poll(
-    cls: type[IO.ComfyNode],
-    request: Luma2GenerationRequest,
-) -> Input.Image:
-    initial = await sync_op(
-        cls,
-        ApiEndpoint(path="/proxy/luma_2/generations", method="POST"),
-        response_model=Luma2Generation,
-        data=request,
-    )
-    if not initial.id:
-        raise RuntimeError("Luma 2 API did not return a generation id.")
-    final = await poll_op(
-        cls,
-        ApiEndpoint(path=f"/proxy/luma_2/generations/{initial.id}", method="GET"),
-        response_model=Luma2Generation,
-        status_extractor=lambda r: r.state,
-        progress_extractor=lambda r: None,
-    )
-    if not final.output:
-        msg = final.failure_reason or "no output returned"
-        raise RuntimeError(f"Luma 2 generation failed: {msg}")
-    url = final.output[0].url
-    if not url:
-        raise RuntimeError("Luma 2 generation completed without an output URL.")
-    return await download_url_to_image_tensor(url)
-
-
-class LumaImageNode(IO.ComfyNode):
-
-    @classmethod
-    def define_schema(cls) -> IO.Schema:
-        return IO.Schema(
-            node_id="LumaImageNode2",
-            display_name="Luma UNI-1 Image",
-            category="api node/image/Luma",
-            description="Generate images from text using the Luma UNI-1 model.",
-            inputs=[
-                IO.String.Input(
-                    "prompt",
-                    multiline=True,
-                    default="",
-                    tooltip="Text description of the desired image. 1–6000 characters.",
-                ),
-                IO.DynamicCombo.Input(
-                    "model",
-                    options=[
-                        IO.DynamicCombo.Option(
-                            "uni-1",
-                            [
-                                IO.Combo.Input(
-                                    "aspect_ratio",
-                                    options=[
-                                        "auto",
-                                        "3:1",
-                                        "2:1",
-                                        "16:9",
-                                        "3:2",
-                                        "1:1",
-                                        "2:3",
-                                        "9:16",
-                                        "1:2",
-                                        "1:3",
-                                    ],
-                                    default="auto",
-                                    tooltip="Output image aspect ratio. 'auto' lets "
-                                    "the model pick based on the prompt.",
-                                ),
-                                *_luma2_uni1_common_inputs(max_image_refs=9),
-                            ],
-                        ),
-                        IO.DynamicCombo.Option(
-                            "uni-1-max",
-                            [
-                                IO.Combo.Input(
-                                    "aspect_ratio",
-                                    options=[
-                                        "auto",
-                                        "3:1",
-                                        "2:1",
-                                        "16:9",
-                                        "3:2",
-                                        "1:1",
-                                        "2:3",
-                                        "9:16",
-                                        "1:2",
-                                        "1:3",
-                                    ],
-                                    default="auto",
-                                    tooltip="Output image aspect ratio. 'auto' lets "
-                                    "the model pick based on the prompt.",
-                                ),
-                                *_luma2_uni1_common_inputs(max_image_refs=9),
-                            ],
-                        ),
-                    ],
-                    tooltip="Model to use for generation.",
-                ),
-                IO.Int.Input(
-                    "seed",
-                    default=0,
-                    min=0,
-                    max=2147483647,
-                    control_after_generate=True,
-                    tooltip="Seed controls whether the node should re-run; "
-                    "results are non-deterministic regardless of seed.",
-                ),
-            ],
-            outputs=[IO.Image.Output()],
-            hidden=[
-                IO.Hidden.auth_token_comfy_org,
-                IO.Hidden.api_key_comfy_org,
-                IO.Hidden.unique_id,
-            ],
-            is_api_node=True,
-            price_badge=IO.PriceBadge(
-                depends_on=IO.PriceBadgeDepends(widgets=["model"], input_groups=["model.image_ref"]),
-                expr="""
-                (
-                  $m := widgets.model;
-                  $refs := $lookup(inputGroups, "model.image_ref");
-                  $base := $m = "uni-1-max" ? 0.1 : 0.0404;
-                  {"type":"usd","usd": $round($base + 0.003 * $refs, 4)}
-                )
-                """,
-            ),
-        )
-
-    @classmethod
-    async def execute(
-        cls,
-        prompt: str,
-        model: dict,
-        seed: int,
-    ) -> IO.NodeOutput:
-        validate_string(prompt, min_length=1, max_length=6000)
-        aspect_ratio = model["aspect_ratio"]
-        style = model["style"]
-        allowed_manga_ratios = {"2:3", "9:16", "1:2", "1:3"}
-        if style == "manga" and aspect_ratio != "auto" and aspect_ratio not in allowed_manga_ratios:
-            raise ValueError(
-                f"'manga' style requires a portrait aspect ratio "
-                f"({', '.join(sorted(allowed_manga_ratios))}) or 'auto'; got '{aspect_ratio}'."
-            )
-        request = Luma2GenerationRequest(
-            prompt=prompt,
-            model=model["model"],
-            type="image",
-            aspect_ratio=aspect_ratio if aspect_ratio != "auto" else None,
-            style=style if style != "auto" else None,
-            output_format="png",
-            web_search=model["web_search"],
-            image_ref=await _luma2_upload_image_refs(cls, model.get("image_ref"), max_count=9),
-        )
-        return IO.NodeOutput(await _luma2_submit_and_poll(cls, request))
-
-
-class LumaImageEditNode(IO.ComfyNode):
-
-    @classmethod
-    def define_schema(cls) -> IO.Schema:
-        return IO.Schema(
-            node_id="LumaImageEditNode2",
-            display_name="Luma UNI-1 Image Edit",
-            category="api node/image/Luma",
-            description="Edit an existing image with a text prompt using the Luma UNI-1 model.",
-            inputs=[
-                IO.Image.Input(
-                    "source",
-                    tooltip="Source image to edit.",
-                ),
-                IO.String.Input(
-                    "prompt",
-                    multiline=True,
-                    default="",
-                    tooltip="Description of the desired edit. 1–6000 characters.",
-                ),
-                IO.DynamicCombo.Input(
-                    "model",
-                    options=[
-                        IO.DynamicCombo.Option(
-                            "uni-1",
-                            _luma2_uni1_common_inputs(max_image_refs=8),
-                        ),
-                        IO.DynamicCombo.Option(
-                            "uni-1-max",
-                            _luma2_uni1_common_inputs(max_image_refs=8),
-                        ),
-                    ],
-                    tooltip="Model to use for editing.",
-                ),
-                IO.Int.Input(
-                    "seed",
-                    default=0,
-                    min=0,
-                    max=2147483647,
-                    control_after_generate=True,
-                    tooltip="Seed controls whether the node should re-run; "
-                    "results are non-deterministic regardless of seed.",
-                ),
-            ],
-            outputs=[IO.Image.Output()],
-            hidden=[
-                IO.Hidden.auth_token_comfy_org,
-                IO.Hidden.api_key_comfy_org,
-                IO.Hidden.unique_id,
-            ],
-            is_api_node=True,
-            price_badge=IO.PriceBadge(
-                depends_on=IO.PriceBadgeDepends(widgets=["model"], input_groups=["model.image_ref"]),
-                expr="""
-                (
-                  $m := widgets.model;
-                  $refs := $lookup(inputGroups, "model.image_ref");
-                  $base := $m = "uni-1-max" ? 0.103 : 0.0434;
-                  {"type":"usd","usd": $round($base + 0.003 * $refs, 4)}
-                )
-                """,
-            ),
-        )
-
-    @classmethod
-    async def execute(
-        cls,
-        source: Input.Image,
-        prompt: str,
-        model: dict,
-        seed: int,
-    ) -> IO.NodeOutput:
-        validate_string(prompt, min_length=1, max_length=6000)
-        request = Luma2GenerationRequest(
-            prompt=prompt,
-            model=model["model"],
-            type="image_edit",
-            source=Luma2ImageRef(url=await upload_image_to_comfyapi(cls, source)),
-            style=model["style"] if model["style"] != "auto" else None,
-            output_format="png",
-            web_search=model["web_search"],
-            image_ref=await _luma2_upload_image_refs(cls, model.get("image_ref"), max_count=8),
-        )
-        return IO.NodeOutput(await _luma2_submit_and_poll(cls, request))
-
-
 class LumaExtension(ComfyExtension):
     @override
     async def get_node_list(self) -> list[type[IO.ComfyNode]]:
@@ -942,8 +654,6 @@ class LumaExtension(ComfyExtension):
             LumaImageToVideoGenerationNode,
             LumaReferenceNode,
             LumaConceptsNode,
-            LumaImageNode,
-            LumaImageEditNode,
         ]
 
 

comfy_api_nodes/nodes_openai.py

@@ -39,18 +39,16 @@ STARTING_POINT_ID_PATTERN = r"<starting_point_id:(.*)>"
 
 
 class SupportedOpenAIModel(str, Enum):
-    gpt_5_5_pro = "gpt-5.5-pro"
-    gpt_5_5 = "gpt-5.5"
-    gpt_5 = "gpt-5"
-    gpt_5_mini = "gpt-5-mini"
-    gpt_5_nano = "gpt-5-nano"
+    o4_mini = "o4-mini"
+    o1 = "o1"
+    o3 = "o3"
+    o1_pro = "o1-pro"
     gpt_4_1 = "gpt-4.1"
     gpt_4_1_mini = "gpt-4.1-mini"
     gpt_4_1_nano = "gpt-4.1-nano"
-    o4_mini = "o4-mini"
-    o3 = "o3"
-    o1_pro = "o1-pro"
-    o1 = "o1"
+    gpt_5 = "gpt-5"
+    gpt_5_mini = "gpt-5-mini"
+    gpt_5_nano = "gpt-5-nano"
 
 
 async def validate_and_cast_response(response, timeout: int = None) -> torch.Tensor:
@@ -741,16 +739,6 @@ class OpenAIChatNode(IO.ComfyNode):
                     "usd": [0.002, 0.008],
                     "format": { "approximate": true, "separator": "-", "suffix": " per 1K tokens" }
                   }
-                  : $contains($m, "gpt-5.5-pro") ? {
-                    "type": "list_usd",
-                    "usd": [0.03, 0.18],
-                    "format": { "approximate": true, "separator": "-", "suffix": " per 1K tokens" }
-                  }
-                  : $contains($m, "gpt-5.5") ? {
-                    "type": "list_usd",
-                    "usd": [0.005, 0.03],
-                    "format": { "approximate": true, "separator": "-", "suffix": " per 1K tokens" }
-                  }
                   : $contains($m, "gpt-5-nano") ? {
                     "type": "list_usd",
                     "usd": [0.00005, 0.0004],

comfy_api_nodes/nodes_sora.py

@@ -33,7 +33,7 @@ class OpenAIVideoSora2(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="OpenAIVideoSora2",
-            display_name="OpenAI Sora - Video (DEPRECATED)",
+            display_name="OpenAI Sora - Video (Deprecated)",
             category="api node/video/Sora",
             description=(
                 "OpenAI video and audio generation.\n\n"

comfy_api_nodes/util/client.py

@@ -19,8 +19,6 @@ from comfy import utils
 from comfy_api.latest import IO
 from server import PromptServer
 
-from comfy.deploy_environment import get_deploy_environment
-
 from . import request_logger
 from ._helpers import (
     default_base_url,
@@ -626,7 +624,6 @@ async def _request_base(cfg: _RequestConfig, expect_binary: bool):
         payload_headers = {"Accept": "*/*"} if expect_binary else {"Accept": "application/json"}
         if not parsed_url.scheme and not parsed_url.netloc:  # is URL relative?
             payload_headers.update(get_auth_header(cfg.node_cls))
-            payload_headers["Comfy-Env"] = get_deploy_environment()
         if cfg.endpoint.headers:
             payload_headers.update(cfg.endpoint.headers)
 

comfy_extras/nodes_ar_video.py

@@ -2,6 +2,7 @@
 ComfyUI nodes for autoregressive video generation (Causal Forcing, Self-Forcing, etc.).
   - EmptyARVideoLatent: create 5D [B, C, T, H, W] video latent tensors
   - SamplerARVideo: SAMPLER for the block-by-block autoregressive denoising loop
+  - ARVideoI2V: image-to-video conditioning for AR models (seeds KV cache with start image)
 """
 
 import torch
@@ -9,6 +10,7 @@ from typing_extensions import override
 
 import comfy.model_management
 import comfy.samplers
+import comfy.utils
 from comfy_api.latest import ComfyExtension, io
 
 
@@ -71,12 +73,62 @@ class SamplerARVideo(io.ComfyNode):
         return io.NodeOutput(comfy.samplers.ksampler("ar_video", extra_options))
 
 
+class ARVideoI2V(io.ComfyNode):
+    """Image-to-video setup for AR video models (Causal Forcing, Self-Forcing).
+
+    VAE-encodes the start image and stores it in the model's transformer_options
+    so that sample_ar_video can seed the KV cache before denoising.
+    Uses the same T2V model checkpoint -- no separate I2V architecture needed.
+    """
+
+    @classmethod
+    def define_schema(cls):
+        return io.Schema(
+            node_id="ARVideoI2V",
+            category="conditioning/video_models",
+            inputs=[
+                io.Model.Input("model"),
+                io.Vae.Input("vae"),
+                io.Image.Input("start_image"),
+                io.Int.Input("width", default=832, min=16, max=8192, step=16),
+                io.Int.Input("height", default=480, min=16, max=8192, step=16),
+                io.Int.Input("length", default=81, min=1, max=1024, step=4),
+                io.Int.Input("batch_size", default=1, min=1, max=64),
+            ],
+            outputs=[
+                io.Model.Output(display_name="MODEL"),
+                io.Latent.Output(display_name="LATENT"),
+            ],
+        )
+
+    @classmethod
+    def execute(cls, model, vae, start_image, width, height, length, batch_size) -> io.NodeOutput:
+        start_image = comfy.utils.common_upscale(
+            start_image[:1].movedim(-1, 1), width, height, "bilinear", "center"
+        ).movedim(1, -1)
+
+        initial_latent = vae.encode(start_image[:, :, :, :3])
+
+        m = model.clone()
+        to = m.model_options.setdefault("transformer_options", {})
+        ar_cfg = to.setdefault("ar_config", {})
+        ar_cfg["initial_latent"] = initial_latent
+
+        lat_t = ((length - 1) // 4) + 1
+        latent = torch.zeros(
+            [batch_size, 16, lat_t, height // 8, width // 8],
+            device=comfy.model_management.intermediate_device(),
+        )
+        return io.NodeOutput(m, {"samples": latent})
+
+
 class ARVideoExtension(ComfyExtension):
     @override
     async def get_node_list(self) -> list[type[io.ComfyNode]]:
         return [
             EmptyARVideoLatent,
             SamplerARVideo,
+            ARVideoI2V,
         ]
 
 

comfy_extras/nodes_frame_interpolation.py

@@ -78,7 +78,7 @@ class FrameInterpolate(io.ComfyNode):
         return io.Schema(
             node_id="FrameInterpolate",
             display_name="Frame Interpolate",
-            category="video",
+            category="image/video",
             search_aliases=["rife", "film", "frame interpolation", "slow motion", "interpolate frames", "vfi"],
             inputs=[
                 FrameInterpolationModel.Input("interp_model"),

comfy_extras/nodes_image_compare.py

@@ -11,7 +11,7 @@ class ImageCompare(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="ImageCompare",
-            display_name="Compare Images",
+            display_name="Image Compare",
             description="Compares two images side by side with a slider.",
             category="image",
             essentials_category="Image Tools",

comfy_extras/nodes_images.py

@@ -24,7 +24,7 @@ class ImageCrop(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageCrop",
             search_aliases=["trim"],
-            display_name="Crop Image (DEPRECATED)",
+            display_name="Image Crop (Deprecated)",
             category="image/transform",
             is_deprecated=True,
             essentials_category="Image Tools",
@@ -56,7 +56,7 @@ class ImageCropV2(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageCropV2",
             search_aliases=["trim"],
-            display_name="Crop Image",
+            display_name="Image Crop",
             category="image/transform",
             essentials_category="Image Tools",
             has_intermediate_output=True,
@@ -109,7 +109,6 @@ class RepeatImageBatch(IO.ComfyNode):
         return IO.Schema(
             node_id="RepeatImageBatch",
             search_aliases=["duplicate image", "clone image"],
-            display_name="Repeat Image Batch",
             category="image/batch",
             inputs=[
                 IO.Image.Input("image"),
@@ -132,7 +131,6 @@ class ImageFromBatch(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageFromBatch",
             search_aliases=["select image", "pick from batch", "extract image"],
-            display_name="Get Image from Batch",
             category="image/batch",
             inputs=[
                 IO.Image.Input("image"),
@@ -159,8 +157,7 @@ class ImageAddNoise(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageAddNoise",
             search_aliases=["film grain"],
-            display_name="Add Noise to Image",
-            category="image/postprocessing",
+            category="image",
             inputs=[
                 IO.Image.Input("image"),
                 IO.Int.Input(
@@ -262,7 +259,7 @@ class ImageStitch(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageStitch",
             search_aliases=["combine images", "join images", "concatenate images", "side by side"],
-            display_name="Stitch Images",
+            display_name="Image Stitch",
             description="Stitches image2 to image1 in the specified direction.\n"
             "If image2 is not provided, returns image1 unchanged.\n"
             "Optional spacing can be added between images.",
@@ -437,7 +434,6 @@ class ResizeAndPadImage(IO.ComfyNode):
         return IO.Schema(
             node_id="ResizeAndPadImage",
             search_aliases=["fit to size"],
-            display_name="Resize And Pad Image",
             category="image/transform",
             inputs=[
                 IO.Image.Input("image"),
@@ -489,7 +485,6 @@ class SaveSVGNode(IO.ComfyNode):
         return IO.Schema(
             node_id="SaveSVGNode",
             search_aliases=["export vector", "save vector graphics"],
-            display_name="Save SVG",
             description="Save SVG files on disk.",
             category="image/save",
             inputs=[
@@ -596,7 +591,7 @@ class ImageRotate(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="ImageRotate",
-            display_name="Rotate Image",
+            display_name="Image Rotate",
             search_aliases=["turn", "flip orientation"],
             category="image/transform",
             essentials_category="Image Tools",
@@ -629,7 +624,6 @@ class ImageFlip(IO.ComfyNode):
         return IO.Schema(
             node_id="ImageFlip",
             search_aliases=["mirror", "reflect"],
-            display_name="Flip Image",
             category="image/transform",
             inputs=[
                 IO.Image.Input("image"),
@@ -656,7 +650,6 @@ class ImageScaleToMaxDimension(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="ImageScaleToMaxDimension",
-            display_name="Scale Image to Max Dimension",
             category="image/upscaling",
             inputs=[
                 IO.Image.Input("image"),
@@ -716,7 +709,7 @@ class SplitImageToTileList(IO.ComfyNode):
     def get_grid_coords(width, height, tile_width, tile_height, overlap):
         coords = []
         stride_x = round(max(tile_width * 0.25, tile_width - overlap))
-        stride_y = round(max(tile_height * 0.25, tile_height - overlap))
+        stride_y = round(max(tile_width * 0.25, tile_height - overlap))
 
         y = 0
         while y < height:

comfy_extras/nodes_lt_audio.py

@@ -147,6 +147,7 @@ class LTXVEmptyLatentAudio(io.ComfyNode):
 
         z_channels = audio_vae.latent_channels
         audio_freq = audio_vae.first_stage_model.latent_frequency_bins
+        sampling_rate = int(audio_vae.first_stage_model.sample_rate)
 
         num_audio_latents = audio_vae.first_stage_model.num_of_latents_from_frames(frames_number, frame_rate)
 
@@ -158,6 +159,7 @@ class LTXVEmptyLatentAudio(io.ComfyNode):
         return io.NodeOutput(
             {
                 "samples": audio_latents,
+                "sample_rate": sampling_rate,
                 "type": "audio",
             }
         )

comfy_extras/nodes_mask.py

@@ -80,8 +80,7 @@ class ImageCompositeMasked(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="ImageCompositeMasked",
-            search_aliases=["overlay", "layer", "paste image", "images composition"],
-            display_name="Image Composite Masked",
+            search_aliases=["paste image", "overlay", "layer"],
             category="image",
             inputs=[
                 IO.Image.Input("destination"),
@@ -202,7 +201,6 @@ class InvertMask(IO.ComfyNode):
         return IO.Schema(
             node_id="InvertMask",
             search_aliases=["reverse mask", "flip mask"],
-            display_name="Invert Mask",
             category="mask",
             inputs=[
                 IO.Mask.Input("mask"),
@@ -224,7 +222,6 @@ class CropMask(IO.ComfyNode):
         return IO.Schema(
             node_id="CropMask",
             search_aliases=["cut mask", "extract mask region", "mask slice"],
-            display_name="Crop Mask",
             category="mask",
             inputs=[
                 IO.Mask.Input("mask"),
@@ -250,8 +247,7 @@ class MaskComposite(IO.ComfyNode):
     def define_schema(cls):
         return IO.Schema(
             node_id="MaskComposite",
-            search_aliases=["combine masks", "blend masks", "layer masks", "masks composition"],
-            display_name="Combine Masks",
+            search_aliases=["combine masks", "blend masks", "layer masks"],
             category="mask",
             inputs=[
                 IO.Mask.Input("destination"),
@@ -302,7 +298,6 @@ class FeatherMask(IO.ComfyNode):
         return IO.Schema(
             node_id="FeatherMask",
             search_aliases=["soft edge mask", "blur mask edges", "gradient mask edge"],
-            display_name="Feather Mask",
             category="mask",
             inputs=[
                 IO.Mask.Input("mask"),

comfy_extras/nodes_morphology.py

@@ -59,8 +59,7 @@ class ImageRGBToYUV(io.ComfyNode):
         return io.Schema(
             node_id="ImageRGBToYUV",
             search_aliases=["color space conversion"],
-            display_name="Image RGB to YUV",
-            category="image/color",
+            category="image/batch",
             inputs=[
                 io.Image.Input("image"),
             ],
@@ -82,8 +81,7 @@ class ImageYUVToRGB(io.ComfyNode):
         return io.Schema(
             node_id="ImageYUVToRGB",
             search_aliases=["color space conversion"],
-            display_name="Image YUV to RGB",
-            category="image/color",
+            category="image/batch",
             inputs=[
                 io.Image.Input("Y"),
                 io.Image.Input("U"),

comfy_extras/nodes_post_processing.py

@@ -20,8 +20,7 @@ class Blend(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="ImageBlend",
-            search_aliases=["mix images"],
-            display_name="Blend Images",
+            display_name="Image Blend",
             category="image/postprocessing",
             essentials_category="Image Tools",
             inputs=[
@@ -225,7 +224,6 @@ class ImageScaleToTotalPixels(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="ImageScaleToTotalPixels",
-            display_name="Scale Image to Total Pixels",
             category="image/upscaling",
             inputs=[
                 io.Image.Input("image"),
@@ -570,7 +568,7 @@ class BatchImagesNode(io.ComfyNode):
         return io.Schema(
             node_id="BatchImagesNode",
             display_name="Batch Images",
-            category="image/batch",
+            category="image",
             essentials_category="Image Tools",
             search_aliases=["batch", "image batch", "batch images", "combine images", "merge images", "stack images"],
             inputs=[

comfy_extras/nodes_primitive.py

@@ -9,8 +9,7 @@ class String(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="PrimitiveString",
-            search_aliases=["text", "string", "text box", "prompt"],
-            display_name="Text String",
+            display_name="String",
             category="utils/primitive",
             inputs=[
                 io.String.Input("value"),
@@ -28,8 +27,7 @@ class StringMultiline(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="PrimitiveStringMultiline",
-            search_aliases=["text", "string", "text multiline", "string multiline", "text box", "prompt"],
-            display_name="Text String (Multiline)",
+            display_name="String (Multiline)",
             category="utils/primitive",
             essentials_category="Basics",
             inputs=[

comfy_extras/nodes_string.py

@@ -10,9 +10,9 @@ class StringConcatenate(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringConcatenate",
-            search_aliases=["concatenate", "text concat", "join text", "merge text", "combine strings", "string concat", "append text", "combine text"],
-            display_name="Concatenate Text",
-            category="text",
+            display_name="Text Concatenate",
+            category="utils/string",
+            search_aliases=["Concatenate", "text concat", "join text", "merge text", "combine strings", "concat", "concatenate", "append text", "combine text", "string"],
             inputs=[
                 io.String.Input("string_a", multiline=True),
                 io.String.Input("string_b", multiline=True),
@@ -33,9 +33,9 @@ class StringSubstring(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringSubstring",
-            search_aliases=["substring", "extract text", "text portion"],
-            display_name="Substring",
-            category="text",
+            search_aliases=["Substring", "extract text", "text portion"],
+            display_name="Text Substring",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.Int.Input("start"),
@@ -58,7 +58,7 @@ class StringLength(io.ComfyNode):
             node_id="StringLength",
             search_aliases=["character count", "text size", "string length"],
             display_name="Text Length",
-            category="text",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
             ],
@@ -77,9 +77,9 @@ class CaseConverter(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="CaseConverter",
-            search_aliases=["case converter", "text case", "uppercase", "lowercase", "capitalize"],
-            display_name="Convert Text Case",
-            category="text",
+            search_aliases=["Case Converter", "text case", "uppercase", "lowercase", "capitalize"],
+            display_name="Text Case Converter",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.Combo.Input("mode", options=["UPPERCASE", "lowercase", "Capitalize", "Title Case"]),
@@ -110,9 +110,9 @@ class StringTrim(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringTrim",
-            search_aliases=["trim", "clean whitespace", "remove whitespace", "remove spaces","strip"],
-            display_name="Trim Text",
-            category="text",
+            search_aliases=["Trim", "clean whitespace", "remove whitespace", "strip"],
+            display_name="Text Trim",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.Combo.Input("mode", options=["Both", "Left", "Right"]),
@@ -141,9 +141,9 @@ class StringReplace(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringReplace",
-            search_aliases=["replace", "find and replace", "substitute", "swap text"],
-            display_name="Replace Text",
-            category="text",
+            search_aliases=["Replace", "find and replace", "substitute", "swap text"],
+            display_name="Text Replace",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.String.Input("find", multiline=True),
@@ -164,9 +164,9 @@ class StringContains(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringContains",
-            search_aliases=["contains", "text includes", "string includes"],
-            display_name="Contains Text",
-            category="text",
+            search_aliases=["Contains", "text includes", "string includes"],
+            display_name="Text Contains",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.String.Input("substring", multiline=True),
@@ -192,9 +192,9 @@ class StringCompare(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="StringCompare",
-            search_aliases=["compare", "text match", "string equals", "starts with", "ends with"],
-            display_name="Compare Text",
-            category="text",
+            search_aliases=["Compare", "text match", "string equals", "starts with", "ends with"],
+            display_name="Text Compare",
+            category="utils/string",
             inputs=[
                 io.String.Input("string_a", multiline=True),
                 io.String.Input("string_b", multiline=True),
@@ -228,9 +228,9 @@ class RegexMatch(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="RegexMatch",
-            search_aliases=["regex match", "regex", "pattern match", "text contains", "string match"],
-            display_name="Match Text",
-            category="text",
+            search_aliases=["Regex Match", "regex", "pattern match", "text contains", "string match"],
+            display_name="Text Match",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.String.Input("regex_pattern", multiline=True),
@@ -269,9 +269,9 @@ class RegexExtract(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="RegexExtract",
-            search_aliases=["regex extract", "regex", "pattern extract", "text parser", "parse text"],
-            display_name="Extract Text",
-            category="text",
+            search_aliases=["Regex Extract", "regex", "pattern extract", "text parser", "parse text"],
+            display_name="Text Extract Substring",
+            category="utils/string",
             inputs=[
                 io.String.Input("string", multiline=True),
                 io.String.Input("regex_pattern", multiline=True),
@@ -344,9 +344,9 @@ class RegexReplace(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="RegexReplace",
-            search_aliases=["regex replace", "regex", "pattern replace", "substitution"],
-            display_name="Replace Text (Regex)",
-            category="text",
+            search_aliases=["Regex Replace", "regex", "pattern replace", "regex replace", "substitution"],
+            display_name="Text Replace (Regex)",
+            category="utils/string",
             description="Find and replace text using regex patterns.",
             inputs=[
                 io.String.Input("string", multiline=True),
@@ -381,8 +381,8 @@ class JsonExtractString(io.ComfyNode):
     def define_schema(cls):
         return io.Schema(
             node_id="JsonExtractString",
-            display_name="Extract Text from JSON",
-            category="text",
+            display_name="Extract String from JSON",
+            category="utils/string",
             search_aliases=["json", "extract json", "parse json", "json value", "read json"],
             inputs=[
                 io.String.Input("json_string", multiline=True),

comfy_extras/nodes_video.py

@@ -17,8 +17,7 @@ class SaveWEBM(io.ComfyNode):
         return io.Schema(
             node_id="SaveWEBM",
             search_aliases=["export webm"],
-            display_name="Save WEBM",
-            category="video",
+            category="image/video",
             is_experimental=True,
             inputs=[
                 io.Image.Input("images"),
@@ -73,7 +72,7 @@ class SaveVideo(io.ComfyNode):
             node_id="SaveVideo",
             search_aliases=["export video"],
             display_name="Save Video",
-            category="video",
+            category="image/video",
             essentials_category="Basics",
             description="Saves the input images to your ComfyUI output directory.",
             inputs=[
@@ -122,7 +121,7 @@ class CreateVideo(io.ComfyNode):
             node_id="CreateVideo",
             search_aliases=["images to video"],
             display_name="Create Video",
-            category="video",
+            category="image/video",
             description="Create a video from images.",
             inputs=[
                 io.Image.Input("images", tooltip="The images to create a video from."),
@@ -147,7 +146,7 @@ class GetVideoComponents(io.ComfyNode):
             node_id="GetVideoComponents",
             search_aliases=["extract frames", "split video", "video to images", "demux"],
             display_name="Get Video Components",
-            category="video",
+            category="image/video",
             description="Extracts all components from a video: frames, audio, and framerate.",
             inputs=[
                 io.Video.Input("video", tooltip="The video to extract components from."),
@@ -175,7 +174,7 @@ class LoadVideo(io.ComfyNode):
             node_id="LoadVideo",
             search_aliases=["import video", "open video", "video file"],
             display_name="Load Video",
-            category="video",
+            category="image/video",
             essentials_category="Basics",
             inputs=[
                 io.Combo.Input("file", options=sorted(files), upload=io.UploadType.video),
@@ -217,7 +216,7 @@ class VideoSlice(io.ComfyNode):
                 "frame load cap",
                 "start time",
             ],
-            category="video",
+            category="image/video",
             essentials_category="Video Tools",
             inputs=[
                 io.Video.Input("video"),

execution.py

@@ -1019,12 +1019,7 @@ async def validate_inputs(prompt_id, prompt, item, validated, visiting=None):
                         combo_options = extra_info.get("options", [])
                     else:
                         combo_options = input_type
-                    is_multiselect = extra_info.get("multiselect", False)
-                    if is_multiselect and isinstance(val, list):
-                        invalid_vals = [v for v in val if v not in combo_options]
-                    else:
-                        invalid_vals = [val] if val not in combo_options else []
-                    if invalid_vals:
+                    if val not in combo_options:
                         input_config = info
                         list_info = ""
 
@@ -1039,7 +1034,7 @@ async def validate_inputs(prompt_id, prompt, item, validated, visiting=None):
                         error = {
                             "type": "value_not_in_list",
                             "message": "Value not in list",
-                            "details": f"{x}: {', '.join(repr(v) for v in invalid_vals)} not in {list_info}",
+                            "details": f"{x}: '{val}' not in {list_info}",
                             "extra_info": {
                                 "input_name": x,
                                 "input_config": input_config,

main.py

@@ -1,21 +1,13 @@
 import comfy.options
 comfy.options.enable_args_parsing()
 
-from comfy.cli_args import args
-
-if args.list_feature_flags:
-    import json
-    from comfy_api.feature_flags import CLI_FEATURE_FLAG_REGISTRY
-    print(json.dumps(CLI_FEATURE_FLAG_REGISTRY, indent=2))  # noqa: T201
-    raise SystemExit(0)
-
 import os
 import importlib.util
 import shutil
 import importlib.metadata
 import folder_paths
 import time
-from comfy.cli_args import enables_dynamic_vram
+from comfy.cli_args import args, enables_dynamic_vram
 from app.logger import setup_logger
 setup_logger(log_level=args.verbose, use_stdout=args.log_stdout)
 

nodes.py

@@ -1887,7 +1887,7 @@ class ImageInvert:
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "invert"
 
-    CATEGORY = "image/color"
+    CATEGORY = "image"
 
     def invert(self, image):
         s = 1.0 - image
@@ -1903,7 +1903,7 @@ class ImageBatch:
     RETURN_TYPES = ("IMAGE",)
     FUNCTION = "batch"
 
-    CATEGORY = "image/batch"
+    CATEGORY = "image"
     DEPRECATED = True
 
     def batch(self, image1, image2):
@@ -1960,7 +1960,7 @@ class ImagePadForOutpaint:
     RETURN_TYPES = ("IMAGE", "MASK")
     FUNCTION = "expand_image"
 
-    CATEGORY = "image/transform"
+    CATEGORY = "image"
 
     def expand_image(self, image, left, top, right, bottom, feathering):
         d1, d2, d3, d4 = image.size()
@@ -2103,7 +2103,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "ConditioningSetArea": "Conditioning (Set Area)",
     "ConditioningSetAreaPercentage": "Conditioning (Set Area with Percentage)",
     "ConditioningSetMask": "Conditioning (Set Mask)",
-    "ControlNetApply": "Apply ControlNet (DEPRECATED)",
+    "ControlNetApply": "Apply ControlNet (OLD)",
     "ControlNetApplyAdvanced": "Apply ControlNet",
     # Latent
     "VAEEncodeForInpaint": "VAE Encode (for Inpainting)",
@@ -2121,7 +2121,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "LatentFromBatch" : "Latent From Batch",
     "RepeatLatentBatch": "Repeat Latent Batch",
     # Image
-    "EmptyImage": "Empty Image",
     "SaveImage": "Save Image",
     "PreviewImage": "Preview Image",
     "LoadImage": "Load Image",
@@ -2129,15 +2128,15 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "LoadImageOutput": "Load Image (from Outputs)",
     "ImageScale": "Upscale Image",
     "ImageScaleBy": "Upscale Image By",
-    "ImageInvert": "Invert Image Colors",
+    "ImageInvert": "Invert Image",
     "ImagePadForOutpaint": "Pad Image for Outpainting",
-    "ImageBatch": "Batch Images (DEPRECATED)",
-    "ImageCrop": "Crop Image",
-    "ImageStitch": "Stitch Images",
-    "ImageBlend": "Blend Images",
-    "ImageBlur": "Blur Image",
-    "ImageQuantize": "Quantize Image",
-    "ImageSharpen": "Sharpen Image",
+    "ImageBatch": "Batch Images",
+    "ImageCrop": "Image Crop",
+    "ImageStitch": "Image Stitch",
+    "ImageBlend": "Image Blend",
+    "ImageBlur": "Image Blur",
+    "ImageQuantize": "Image Quantize",
+    "ImageSharpen": "Image Sharpen",
     "ImageScaleToTotalPixels": "Scale Image to Total Pixels",
     "GetImageSize": "Get Image Size",
     # _for_testing
@@ -2262,7 +2261,7 @@ async def load_custom_node(module_path: str, ignore=set(), module_parent="custom
                 logging.warning(f"Error while calling comfy_entrypoint in {module_path}: {e}")
                 return False
         else:
-            logging.warning(f"Skip {module_path} module for custom nodes due to the lack of NODE_CLASS_MAPPINGS or comfy_entrypoint (need one).")
+            logging.warning(f"Skip {module_path} module for custom nodes due to the lack of NODE_CLASS_MAPPINGS or NODES_LIST (need one).")
             return False
     except Exception as e:
         logging.warning(traceback.format_exc())

openapi.yaml

@@ -1999,26 +1999,6 @@ components:
           items:
             type: string
           description: List of node IDs to execute (partial graph execution)
-        workflow_id:
-          type: string
-          format: uuid
-          nullable: true
-          x-runtime: [cloud]
-          description: |
-            UUID identifying a hosted-cloud workflow entity to associate with this
-            job. Local ComfyUI doesn't track workflow entities and returns `null`
-            (or omits the field). The `x-runtime: [cloud]` extension marks this
-            as populated only by the hosted-cloud runtime; absence of the tag
-            means a field is populated by all runtimes.
-        workflow_version_id:
-          type: string
-          format: uuid
-          nullable: true
-          x-runtime: [cloud]
-          description: |
-            UUID identifying a hosted-cloud workflow version to associate with
-            this job. Local ComfyUI returns `null` (or omits the field). See
-            `workflow_id` above for `x-runtime` semantics.
 
     PromptResponse:
       type: object
@@ -2367,12 +2347,7 @@ components:
           description: Device type (cuda, mps, cpu, etc.)
         index:
           type: number
-          nullable: true
-          description: |
-            Device index within its type (e.g. CUDA ordinal for `cuda:0`,
-            `cuda:1`). `null` for devices with no index, including the CPU
-            device returned in `--cpu` mode (PyTorch's `torch.device('cpu').index`
-            is `None`).
+          description: Device index
         vram_total:
           type: number
           description: Total VRAM in bytes
@@ -2528,18 +2503,7 @@ components:
           description: Alternative search terms for finding this node
         essentials_category:
           type: string
-          nullable: true
-          description: |
-            Category override used by the essentials pack. The
-            `essentials_category` key may be present with a string value,
-            present and `null`, or absent entirely:
-
-            - V1 nodes: `essentials_category` is **omitted** when the node
-              class doesn't define an `ESSENTIALS_CATEGORY` attribute, and
-              **`null`** if the attribute is explicitly set to `None`.
-            - V3 nodes (`comfy_api.latest.io`): `essentials_category` is
-              **always present**, and **`null`** for nodes whose `Schema`
-              doesn't populate it.
+          description: Category override used by the essentials pack
 
     # -------------------------------------------------------------------
     # Models

requirements.txt

@@ -1,5 +1,5 @@
 comfyui-frontend-package==1.42.15
-comfyui-workflow-templates==0.9.69
+comfyui-workflow-templates==0.9.68
 comfyui-embedded-docs==0.4.4
 torch
 torchsde

server.py

@@ -560,7 +560,7 @@ class PromptServer():
                             buffer.seek(0)
 
                             return web.Response(body=buffer.read(), content_type=f'image/{image_format}',
-                                                headers={"Content-Disposition": f"attachment; filename=\"{filename}\""})
+                                                headers={"Content-Disposition": f"filename=\"{filename}\""})
 
                     if 'channel' not in request.rel_url.query:
                         channel = 'rgba'
@@ -580,7 +580,7 @@ class PromptServer():
                             buffer.seek(0)
 
                             return web.Response(body=buffer.read(), content_type='image/png',
-                                                headers={"Content-Disposition": f"attachment; filename=\"{filename}\""})
+                                                headers={"Content-Disposition": f"filename=\"{filename}\""})
 
                     elif channel == 'a':
                         with Image.open(file) as img:
@@ -597,7 +597,7 @@ class PromptServer():
                             alpha_buffer.seek(0)
 
                             return web.Response(body=alpha_buffer.read(), content_type='image/png',
-                                                headers={"Content-Disposition": f"attachment; filename=\"{filename}\""})
+                                                headers={"Content-Disposition": f"filename=\"{filename}\""})
                     else:
                         # Use the content type from asset resolution if available,
                         # otherwise guess from the filename.
@@ -614,7 +614,7 @@ class PromptServer():
                         return web.FileResponse(
                             file,
                             headers={
-                                "Content-Disposition": f"attachment; filename=\"{filename}\"",
+                                "Content-Disposition": f"filename=\"{filename}\"",
                                 "Content-Type": content_type
                             }
                         )

tests-unit/comfy_api_test/multicombo_serialization_test.py

@@ -1,78 +0,0 @@
-from comfy_api.latest._io import Combo, MultiCombo
-
-
-def test_multicombo_serializes_multi_select_as_object():
-    multi_combo = MultiCombo.Input(
-        id="providers",
-        options=["a", "b", "c"],
-        default=["a"],
-    )
-
-    serialized = multi_combo.as_dict()
-
-    assert serialized["multiselect"] is True
-    assert "multi_select" in serialized
-    assert serialized["multi_select"] == {}
-
-
-def test_multicombo_serializes_multi_select_with_placeholder_and_chip():
-    multi_combo = MultiCombo.Input(
-        id="providers",
-        options=["a", "b", "c"],
-        default=["a"],
-        placeholder="Select providers",
-        chip=True,
-    )
-
-    serialized = multi_combo.as_dict()
-
-    assert serialized["multiselect"] is True
-    assert serialized["multi_select"] == {
-        "placeholder": "Select providers",
-        "chip": True,
-    }
-
-
-def test_combo_does_not_serialize_multiselect():
-    """Regular Combo should not have multiselect in its serialized output."""
-    combo = Combo.Input(
-        id="choice",
-        options=["a", "b", "c"],
-    )
-
-    serialized = combo.as_dict()
-
-    # Combo sets multiselect=False, but prune_dict keeps False (not None),
-    # so it should be present but False
-    assert serialized.get("multiselect") is False
-    assert "multi_select" not in serialized
-
-
-def _validate_combo_values(val, combo_options, is_multiselect):
-    """Reproduce the validation logic from execution.py for testing."""
-    if is_multiselect and isinstance(val, list):
-        return [v for v in val if v not in combo_options]
-    else:
-        return [val] if val not in combo_options else []
-
-
-def test_multicombo_validation_accepts_valid_list():
-    options = ["a", "b", "c"]
-    assert _validate_combo_values(["a", "b"], options, True) == []
-
-
-def test_multicombo_validation_rejects_invalid_values():
-    options = ["a", "b", "c"]
-    assert _validate_combo_values(["a", "x"], options, True) == ["x"]
-
-
-def test_multicombo_validation_accepts_empty_list():
-    options = ["a", "b", "c"]
-    assert _validate_combo_values([], options, True) == []
-
-
-def test_combo_validation_rejects_list_even_with_valid_items():
-    """A regular Combo should not accept a list value."""
-    options = ["a", "b", "c"]
-    invalid = _validate_combo_values(["a", "b"], options, False)
-    assert len(invalid) > 0

tests-unit/deploy_environment_test.py

@@ -1,109 +0,0 @@
-"""Tests for comfy.deploy_environment."""
-
-import os
-
-import pytest
-
-from comfy import deploy_environment
-from comfy.deploy_environment import get_deploy_environment
-
-
-@pytest.fixture(autouse=True)
-def _reset_cache_and_install_dir(tmp_path, monkeypatch):
-    """Reset the functools cache and point the ComfyUI install dir at a tmp dir for each test."""
-    get_deploy_environment.cache_clear()
-    monkeypatch.setattr(deploy_environment, "_COMFY_INSTALL_DIR", str(tmp_path))
-    yield
-    get_deploy_environment.cache_clear()
-
-
-def _write_env_file(tmp_path, content: str) -> str:
-    """Write the env file with exact content (no newline translation).
-
-    `newline=""` disables Python's text-mode newline translation so the bytes
-    on disk match the literal string passed in, regardless of host OS.
-    Newline-style tests (CRLF, lone CR) rely on this.
-    """
-    path = os.path.join(str(tmp_path), ".comfy_environment")
-    with open(path, "w", encoding="utf-8", newline="") as f:
-        f.write(content)
-    return path
-
-
-class TestGetDeployEnvironment:
-    def test_returns_local_git_when_file_missing(self):
-        assert get_deploy_environment() == "local-git"
-
-    def test_reads_value_from_file(self, tmp_path):
-        _write_env_file(tmp_path, "local-desktop2-standalone\n")
-        assert get_deploy_environment() == "local-desktop2-standalone"
-
-    def test_strips_trailing_whitespace_and_newline(self, tmp_path):
-        _write_env_file(tmp_path, "  local-desktop2-standalone  \n")
-        assert get_deploy_environment() == "local-desktop2-standalone"
-
-    def test_only_first_line_is_used(self, tmp_path):
-        _write_env_file(tmp_path, "first-line\nsecond-line\n")
-        assert get_deploy_environment() == "first-line"
-
-    def test_crlf_line_ending(self, tmp_path):
-        # Windows editors often save text files with CRLF line endings.
-        # The CR must not end up in the returned value.
-        _write_env_file(tmp_path, "local-desktop2-standalone\r\n")
-        assert get_deploy_environment() == "local-desktop2-standalone"
-
-    def test_crlf_multiline_only_first_line_used(self, tmp_path):
-        _write_env_file(tmp_path, "first-line\r\nsecond-line\r\n")
-        assert get_deploy_environment() == "first-line"
-
-    def test_crlf_with_surrounding_whitespace(self, tmp_path):
-        _write_env_file(tmp_path, "  local-desktop2-standalone  \r\n")
-        assert get_deploy_environment() == "local-desktop2-standalone"
-
-    def test_lone_cr_line_ending(self, tmp_path):
-        # Classic-Mac / some legacy editors use a bare CR.
-        # Universal-newlines decoding treats it as a line terminator too.
-        _write_env_file(tmp_path, "local-desktop2-standalone\r")
-        assert get_deploy_environment() == "local-desktop2-standalone"
-
-    def test_empty_file_falls_back_to_default(self, tmp_path):
-        _write_env_file(tmp_path, "")
-        assert get_deploy_environment() == "local-git"
-
-    def test_empty_after_whitespace_strip_falls_back_to_default(self, tmp_path):
-        _write_env_file(tmp_path, "   \n")
-        assert get_deploy_environment() == "local-git"
-
-    def test_strips_control_chars_within_first_line(self, tmp_path):
-        # Embedded NUL/control chars in the value should be stripped
-        # (header-injection / smuggling protection).
-        _write_env_file(tmp_path, "abc\x00\x07xyz\n")
-        assert get_deploy_environment() == "abcxyz"
-
-    def test_strips_non_ascii_characters(self, tmp_path):
-        _write_env_file(tmp_path, "café-é\n")
-        assert get_deploy_environment() == "caf-"
-
-    def test_caps_read_at_128_bytes(self, tmp_path):
-        # A single huge line with no newline must not be fully read into memory.
-        huge = "x" * 10_000
-        _write_env_file(tmp_path, huge)
-        result = get_deploy_environment()
-        assert result == "x" * 128
-
-    def test_result_is_cached_across_calls(self, tmp_path):
-        path = _write_env_file(tmp_path, "first_value\n")
-        assert get_deploy_environment() == "first_value"
-        # Overwrite the file — cached value should still be returned.
-        with open(path, "w", encoding="utf-8") as f:
-            f.write("second_value\n")
-        assert get_deploy_environment() == "first_value"
-
-    def test_unreadable_file_falls_back_to_default(self, tmp_path, monkeypatch):
-        _write_env_file(tmp_path, "should_not_be_used\n")
-
-        def _boom(*args, **kwargs):
-            raise OSError("simulated read failure")
-
-        monkeypatch.setattr("builtins.open", _boom)
-        assert get_deploy_environment() == "local-git"

tests-unit/feature_flags_test.py

@@ -1,15 +1,10 @@
 """Tests for feature flags functionality."""
 
-import pytest
-
 from comfy_api.feature_flags import (
     get_connection_feature,
     supports_feature,
     get_server_features,
-    CLI_FEATURE_FLAG_REGISTRY,
     SERVER_FEATURE_FLAGS,
-    _coerce_flag_value,
-    _parse_cli_feature_flags,
 )
 
 
@@ -101,83 +96,3 @@ class TestFeatureFlags:
         result = get_connection_feature(sockets_metadata, "sid1", "any_feature")
         assert result is False
         assert supports_feature(sockets_metadata, "sid1", "any_feature") is False
-
-
-class TestCoerceFlagValue:
-    """Test suite for _coerce_flag_value."""
-
-    def test_registered_bool_true(self):
-        assert _coerce_flag_value("show_signin_button", "true") is True
-        assert _coerce_flag_value("show_signin_button", "True") is True
-
-    def test_registered_bool_false(self):
-        assert _coerce_flag_value("show_signin_button", "false") is False
-        assert _coerce_flag_value("show_signin_button", "FALSE") is False
-
-    def test_unregistered_key_stays_string(self):
-        assert _coerce_flag_value("unknown_flag", "true") == "true"
-        assert _coerce_flag_value("unknown_flag", "42") == "42"
-
-    def test_bool_typo_raises(self):
-        """Strict bool: typos like 'ture' or 'yes' must raise so the flag can be dropped."""
-        with pytest.raises(ValueError):
-            _coerce_flag_value("show_signin_button", "ture")
-        with pytest.raises(ValueError):
-            _coerce_flag_value("show_signin_button", "yes")
-        with pytest.raises(ValueError):
-            _coerce_flag_value("show_signin_button", "1")
-        with pytest.raises(ValueError):
-            _coerce_flag_value("show_signin_button", "")
-
-    def test_failed_int_coercion_raises(self, monkeypatch):
-        """Malformed values for typed flags must raise; caller decides what to do."""
-        monkeypatch.setitem(
-            CLI_FEATURE_FLAG_REGISTRY,
-            "test_int_flag",
-            {"type": "int", "default": 0, "description": "test"},
-        )
-        with pytest.raises(ValueError):
-            _coerce_flag_value("test_int_flag", "not_a_number")
-
-
-class TestParseCliFeatureFlags:
-    """Test suite for _parse_cli_feature_flags."""
-
-    def test_single_flag(self, monkeypatch):
-        monkeypatch.setattr("comfy_api.feature_flags.args", type("Args", (), {"feature_flag": ["show_signin_button=true"]})())
-        result = _parse_cli_feature_flags()
-        assert result == {"show_signin_button": True}
-
-    def test_missing_equals_defaults_to_true(self, monkeypatch):
-        """Bare flag without '=' is treated as the string 'true' (and coerced if registered)."""
-        monkeypatch.setattr("comfy_api.feature_flags.args", type("Args", (), {"feature_flag": ["show_signin_button", "valid=1"]})())
-        result = _parse_cli_feature_flags()
-        assert result == {"show_signin_button": True, "valid": "1"}
-
-    def test_empty_key_skipped(self, monkeypatch):
-        monkeypatch.setattr("comfy_api.feature_flags.args", type("Args", (), {"feature_flag": ["=value", "valid=1"]})())
-        result = _parse_cli_feature_flags()
-        assert result == {"valid": "1"}
-
-    def test_invalid_bool_value_dropped(self, monkeypatch, caplog):
-        """A typo'd bool value must be dropped entirely, not silently set to False
-        and not stored as a raw string. A warning must be logged."""
-        monkeypatch.setattr(
-            "comfy_api.feature_flags.args",
-            type("Args", (), {"feature_flag": ["show_signin_button=ture", "valid=1"]})(),
-        )
-        with caplog.at_level("WARNING"):
-            result = _parse_cli_feature_flags()
-        assert result == {"valid": "1"}
-        assert "show_signin_button" not in result
-        assert any("show_signin_button" in r.message and "drop" in r.message.lower() for r in caplog.records)
-
-
-class TestCliFeatureFlagRegistry:
-    """Test suite for the CLI feature flag registry."""
-
-    def test_registry_entries_have_required_fields(self):
-        for key, info in CLI_FEATURE_FLAG_REGISTRY.items():
-            assert "type" in info, f"{key} missing 'type'"
-            assert "default" in info, f"{key} missing 'default'"
-            assert "description" in info, f"{key} missing 'description'"