TypeError: patched_forward_orig() got an unexpected keyword argument 'timestep_zero_index'

[jerocabezasgallego-bit]

Custom Node Testing

• I have tried disabling custom nodes and the issue persists (see how to disable custom nodes if you need help)

Your question

I'm experimenting an error with a workflow that worked fine yesterday. I've tested to disable custom nodes and its still there.

System Info
OS
win32
Python Version
3.13.11 (tags/v3.13.11:6278944, Dec 5 2025, 16:26:58) [MSC v.1944 64 bit (AMD64)]
Embedded Python
true
Pytorch Version
2.10.0+cu130
Arguments
ComfyUI\main.py --windows-standalone-build
RAM Total
127.65 GB
RAM Free
110.26 GB
Templates Version
0.9.26
Name
cuda:0 NVIDIA RTX A4500 : cudaMallocAsync
Type
cuda
VRAM Total
19.99 GB
VRAM Free
18.85 GB

Logs

D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI
** ComfyUI Base Folder Path: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI
** User directory: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user
** ComfyUI-Manager config path: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user\__manager\config.ini
** Log path: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user\comfyui.log

Prestartup times for custom nodes:
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-easy-use
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\rgthree-comfy
   6.9 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-manager

D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\cuda\__init__.py:65: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you.
  import pynvml  # type: ignore[import]
Found comfy_kitchen backend eager: {'available': True, 'disabled': False, 'unavailable_reason': None, 'capabilities': ['apply_rope', 'apply_rope1', 'dequantize_mxfp8', 'dequantize_nvfp4', 'dequantize_per_tensor_fp8', 'quantize_mxfp8', 'quantize_nvfp4', 'quantize_per_tensor_fp8', 'scaled_mm_mxfp8', 'scaled_mm_nvfp4']}
Found comfy_kitchen backend triton: {'available': False, 'disabled': True, 'unavailable_reason': "ImportError: No module named 'triton'", 'capabilities': []}
Found comfy_kitchen backend cuda: {'available': True, 'disabled': False, 'unavailable_reason': None, 'capabilities': ['apply_rope', 'apply_rope1', 'dequantize_nvfp4', 'dequantize_per_tensor_fp8', 'quantize_mxfp8', 'quantize_nvfp4', 'quantize_per_tensor_fp8', 'scaled_mm_nvfp4']}
Checkpoint files will always be loaded safely.
Total VRAM 20470 MB, total RAM 130711 MB
pytorch version: 2.10.0+cu130
Set vram state to: NORMAL_VRAM
Device: cuda:0 NVIDIA RTX A4500 : cudaMallocAsync
Using async weight offloading with 2 streams
Enabled pinned memory 58820.0
working around nvidia conv3d memory bug.
Using pytorch attention
DynamicVRAM support detected and enabled
Python version: 3.13.11 (tags/v3.13.11:6278944, Dec  5 2025, 16:26:58) [MSC v.1944 64 bit (AMD64)]
ComfyUI version: 0.18.1
comfy-aimdo version: 0.2.12
comfy-kitchen version: 0.2.8
ComfyUI frontend version: 1.41.21
[Prompt Server] web root: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\comfyui_frontend_package\static
Asset seeder disabled
[ComfyUI-Easy-Use] server: v1.3.6 Loaded
[ComfyUI-Easy-Use] web root: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-easy-use\web_version/v2 Loaded
[Flux2 Fun] ControlNet patch applied
ComfyUI-GGUF: Allowing full torch compile
Traceback (most recent call last):
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\nodes.py", line 2227, in load_custom_node
    module_spec.loader.exec_module(module)
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^
  File "<frozen importlib._bootstrap_external>", line 1023, in exec_module
  File "<frozen importlib._bootstrap>", line 488, in _call_with_frames_removed
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-GGUF-FantasyTalking\__init__.py", line 7, in <module>
    from .nodes import NODE_CLASS_MAPPINGS
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-GGUF-FantasyTalking\nodes.py", line 608
    """
    ^^^
SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes in position 149-150: truncated \UXXXXXXXX escape

Cannot import D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-GGUF-FantasyTalking module for custom nodes: (unicode error) 'unicodeescape' codec can't decode bytes in position 149-150: truncated \UXXXXXXXX escape (nodes.py, line 608)
### Loading: ComfyUI-Impact-Pack (V8.28.2)
[Impact Pack] Wildcard total size (0.00 MB) is within cache limit (50.00 MB). Using full cache mode.
[Impact Pack] Wildcards loading done.
### Loading: ComfyUI-Impact-Subpack (V1.3.5)
[Impact Pack/Subpack] Using folder_paths to determine whitelist path: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user\default\ComfyUI-Impact-Subpack\model-whitelist.txt
[Impact Pack/Subpack] Ensured whitelist directory exists: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user\default\ComfyUI-Impact-Subpack
[Impact Pack/Subpack] Loaded 0 model(s) from whitelist: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\user\default\ComfyUI-Impact-Subpack\model-whitelist.txt
[Impact Subpack] ultralytics_bbox: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\models\ultralytics\bbox
[Impact Subpack] ultralytics_segm: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\models\ultralytics\segm
theUpsiders Logic Nodes: Loaded
### Loading: ComfyUI-Manager (V3.39.2)
[ComfyUI-Manager] network_mode: public
[ComfyUI-Manager] ComfyUI per-queue preview override detected (PR #11261). Manager's preview method feature is disabled. Use ComfyUI's --preview-method CLI option or 'Settings > Execution > Live preview method'.
### ComfyUI Revision: 362 [ebf6b52e] *DETACHED | Released on '2026-03-21'
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/alter-list.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/github-stats.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/model-list.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/extension-node-map.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json
[UmeAiRT-Toolkit] Logic Nodes: Could not import FaceDetailer internals: No module named 'D:\\00_jero\\software_temp\\ComfyUI\\ComfyUI_windows_portable\\ComfyUI\\custom_nodes\\ComfyUI-UmeAiRT-Toolkit.modules.facedetailer_core'
[UmeAiRT-Toolkit] 🧩 Loaded 47 nodes.
[UmeAiRT-Toolkit] ⚡ Optimisation check: SageAttention ❌ | Flash Attention ❌ | Triton ❌
[UmeAiRT-Toolkit] 💡 Optional: pip install flash-attn
[UmeAiRT-Toolkit] 📊 Initial CUDA memory: 18.85GB free / 19.99GB total
[UmeAiRT-Toolkit] ✅ Initialization Complete.
ComfyUI-GGUF: Allowing full torch compile
------------------------------------------
Comfyroll Studio v1.76 :  175 Nodes Loaded
------------------------------------------
** For changes, please see patch notes at https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/blob/main/Patch_Notes.md
** For help, please see the wiki at https://github.com/Suzie1/ComfyUI_Comfyroll_CustomNodes/wiki
------------------------------------------
[D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using ckpts path: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux\ckpts
[D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using symlinks: False
[D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using ort providers: ['CUDAExecutionProvider', 'DirectMLExecutionProvider', 'OpenVINOExecutionProvider', 'ROCMExecutionProvider', 'MIGraphXExecutionProvider', 'CPUExecutionProvider', 'CoreMLExecutionProvider']
D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux\node_wrappers\dwpose.py:26: UserWarning: DWPose: Onnxruntime not found or doesn't come with acceleration providers, switch to OpenCV with CPU device. DWPose might run very slowly
  warnings.warn("DWPose: Onnxruntime not found or doesn't come with acceleration providers, switch to OpenCV with CPU device. DWPose might run very slowly")
USDU batch patches applied successfully.
USDU batch patches applied successfully.
Efficiency Nodes: Attempting to add Control Net options to the 'HiRes-Fix Script' Node (comfyui_controlnet_aux add-on)...Success!

[rgthree-comfy] Loaded 48 magnificent nodes. 🎉

[rgthree-comfy] ComfyUI's new Node 2.0 rendering may be incompatible with some rgthree-comfy nodes and features, breaking some rendering as well as losing the ability to access a node's properties (a vital part of many nodes). It also appears to run MUCH more slowly spiking CPU usage and causing jankiness and unresponsiveness, especially with large workflows. Personally I am not planning to use the new Nodes 2.0 and, unfortunately, am not able to invest the time to investigate and overhaul rgthree-comfy where needed. If you have issues when Nodes 2.0 is enabled, I'd urge you to switch it off as well and join me in hoping ComfyUI is not planning to deprecate the existing, stable canvas rendering all together.

FETCH ComfyRegistry Data: 5/134
WAS Node Suite: OpenCV Python FFMPEG support is enabled
WAS Node Suite Warning: `ffmpeg_bin_path` is not set in `D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\was-node-suite-comfyui\was_suite_config.json` config file. Will attempt to use system ffmpeg binaries if available.
WAS Node Suite: Finished. Loaded 220 nodes successfully.

        "Art is the daughter of freedom." - Friedrich Schiller


Import times for custom nodes:
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\websocket_image_save.py
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-image-compare
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Alimama-ControlNet-compatible
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\wlsh_nodes
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-logic
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-flux2fun-controlnet
   0.0 seconds (IMPORT FAILED): D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-GGUF-FantasyTalking
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-inpaint-nodes
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-TiledDiffusion
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\masquerade-nodes-comfyui
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-post-processing-nodes
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-GGUF
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_ipadapter_plus
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\cg-image-filter
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_essentials
   0.0 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Zlycoris
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_ultimatesdupscale
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-kjnodes
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-segment-anything-2
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\rgthree-comfy
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-UmeAiRT-Toolkit
   0.1 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI_Comfyroll_CustomNodes
   0.2 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\x-flux-comfyui
   0.2 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\efficiency-nodes-comfyui
   0.2 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui_controlnet_aux
   0.2 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\kaytool
   0.3 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-impact-subpack
   0.5 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-manager
   0.5 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-impact-pack
   0.9 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-florence2
   2.8 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-tensorops
   2.9 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\was-node-suite-comfyui
   5.8 seconds: D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\comfyui-easy-use

Context impl SQLiteImpl.
Will assume non-transactional DDL.
Starting server

To see the GUI go to: http://127.0.0.1:8188
FETCH ComfyRegistry Data: 10/134
[DEPRECATION WARNING] Detected import of deprecated legacy API: /scripts/ui.js. This is likely caused by a custom node extension using outdated APIs. Please update your extensions or contact the extension author for an updated version.
[DEPRECATION WARNING] Detected import of deprecated legacy API: /extensions/core/clipspace.js. This is likely caused by a custom node extension using outdated APIs. Please update your extensions or contact the extension author for an updated version.
[DEPRECATION WARNING] Detected import of deprecated legacy API: /extensions/core/groupNode.js. This is likely caused by a custom node extension using outdated APIs. Please update your extensions or contact the extension author for an updated version.
ComfyUI-GGUF: Allowing full torch compile
[UmeAiRT-Toolkit] SeedVR2: Using vendored copy
[]
[]
FETCH ComfyRegistry Data: 15/134
[DEPRECATION WARNING] Detected import of deprecated legacy API: /scripts/ui/components/buttonGroup.js. This is likely caused by a custom node extension using outdated APIs. Please update your extensions or contact the extension author for an updated version.
[DEPRECATION WARNING] Detected import of deprecated legacy API: /scripts/ui/components/button.js. This is likely caused by a custom node extension using outdated APIs. Please update your extensions or contact the extension author for an updated version.
FETCH ComfyRegistry Data: 20/134
FETCH ComfyRegistry Data: 25/134
FETCH ComfyRegistry Data: 30/134
FETCH ComfyRegistry Data: 35/134
FETCH ComfyRegistry Data: 40/134
FETCH ComfyRegistry Data: 45/134
FETCH ComfyRegistry Data: 50/134
FETCH ComfyRegistry Data: 55/134
FETCH ComfyRegistry Data: 60/134
FETCH ComfyRegistry Data: 65/134
FETCH ComfyRegistry Data: 70/134
FETCH ComfyRegistry Data: 75/134
got prompt
Using pytorch attention in VAE
Using pytorch attention in VAE
VAE load device: cuda:0, offload device: cpu, dtype: torch.bfloat16
Requested to load AutoencoderKL
Model AutoencoderKL prepared for dynamic VRAM loading. 160MB Staged. 0 patches attached.
FETCH ComfyRegistry Data: 80/134
Found quantization metadata version 1
Using MixedPrecisionOps for text encoder
CLIP/text encoder model load device: cuda:0, offload device: cpu, current: cpu, dtype: torch.float16
Requested to load Flux2TEModel_
Model Flux2TEModel_ prepared for dynamic VRAM loading. 8262MB Staged. 0 patches attached. Force pre-loaded 145 weights: 602 KB.
FETCH ComfyRegistry Data: 85/134
FETCH ComfyRegistry Data: 90/134
Found quantization metadata version 1
Detected mixed precision quantization
Using mixed precision operations
model weight dtype torch.bfloat16, manual cast: torch.bfloat16
model_type FLUX
Requested to load Flux2
Model Flux2 prepared for dynamic VRAM loading. 8996MB Staged. 0 patches attached.
  0%|                                                                | 0/4 [00:00<?, ?it/s,   Model Initializing ...  ]
!!! Exception during processing !!! patched_forward_orig() got an unexpected keyword argument 'timestep_zero_index'
Traceback (most recent call last):
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\execution.py", line 525, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
                                                              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\execution.py", line 334, in get_output_data
    return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\execution.py", line 308, in _async_map_node_over_list
    await process_inputs(input_dict, i)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\execution.py", line 296, in process_inputs
    result = f(**inputs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy_api\internal\__init__.py", line 149, in wrapped_func
    return method(locked_class, **inputs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy_api\latest\_io.py", line 1764, in EXECUTE_NORMALIZED
    to_return = cls.execute(*args, **kwargs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy_extras\nodes_custom_sampler.py", line 963, in execute
    samples = guider.sample(noise.generate_noise(latent), latent_image, sampler, sigmas, denoise_mask=noise_mask, callback=callback, disable_pbar=disable_pbar, seed=noise.seed)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 1052, in sample
    output = executor.execute(noise, latent_image, sampler, sigmas, denoise_mask, callback, disable_pbar, seed, latent_shapes=latent_shapes)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 995, in outer_sample
    output = self.inner_sample(noise, latent_image, device, sampler, sigmas, denoise_mask, callback, disable_pbar, seed, latent_shapes=latent_shapes)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 981, in inner_sample
    samples = executor.execute(self, sigmas, extra_args, callback, noise, latent_image, denoise_mask, disable_pbar)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-TiledDiffusion\utils.py", line 34, in KSAMPLER_sample
    return orig_fn(*args, **kwargs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 751, in sample
    samples = self.sampler_function(model_k, noise, sigmas, extra_args=extra_args, callback=k_callback, disable=disable_pbar, **self.extra_options)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\utils\_contextlib.py", line 124, in decorate_context
    return func(*args, **kwargs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\k_diffusion\sampling.py", line 205, in sample_euler
    denoised = model(x, sigma_hat * s_in, **extra_args)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 400, in __call__
    out = self.inner_model(x, sigma, model_options=model_options, seed=seed)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 954, in __call__
    return self.outer_predict_noise(*args, **kwargs)
           ~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 961, in outer_predict_noise
    ).execute(x, timestep, model_options, seed)
      ~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 964, in predict_noise
    return sampling_function(self.inner_model, x, timestep, self.conds.get("negative", None), self.conds.get("positive", None), self.cfg, model_options=model_options, seed=seed)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 380, in sampling_function
    out = calc_cond_batch(model, conds, x, timestep, model_options)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 205, in calc_cond_batch
    return _calc_cond_batch_outer(model, conds, x_in, timestep, model_options)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 213, in _calc_cond_batch_outer
    return executor.execute(model, conds, x_in, timestep, model_options)
           ~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\samplers.py", line 325, in _calc_cond_batch
    output = model.apply_model(input_x, timestep_, **c).chunk(batch_chunks)
             ~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\model_base.py", line 171, in apply_model
    return comfy.patcher_extension.WrapperExecutor.new_class_executor(
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    ...<2 lines>...
        comfy.patcher_extension.get_all_wrappers(comfy.patcher_extension.WrappersMP.APPLY_MODEL, transformer_options)
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    ).execute(x, t, c_concat, c_crossattn, control, transformer_options, **kwargs)
    ~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\model_base.py", line 210, in _apply_model
    model_output = self.diffusion_model(xc, t, context=context, control=control, transformer_options=transformer_options, **extra_conds)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\nn\modules\module.py", line 1776, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\nn\modules\module.py", line 1787, in _call_impl
    return forward_call(*args, **kwargs)
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\ldm\flux\model.py", line 345, in forward
    return comfy.patcher_extension.WrapperExecutor.new_class_executor(
           ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    ...<2 lines>...
        comfy.patcher_extension.get_all_wrappers(comfy.patcher_extension.WrappersMP.DIFFUSION_MODEL, transformer_options)
        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
    ).execute(x, timestep, context, y, guidance, ref_latents, control, transformer_options, **kwargs)
    ~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\patcher_extension.py", line 112, in execute
    return self.original(*args, **kwargs)
           ~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "D:\00_jero\software_temp\ComfyUI\ComfyUI_windows_portable\ComfyUI\comfy\ldm\flux\model.py", line 406, in _forward
    out = self.forward_orig(img, img_ids, context, txt_ids, timestep, y, guidance, control, timestep_zero_index=timestep_zero_index, transformer_options=transformer_options, attn_mask=kwargs.get("attention_mask", None))
TypeError: patched_forward_orig() got an unexpected keyword argument 'timestep_zero_index'

Prompt executed in 9.35 seconds
FETCH ComfyRegistry Data: 95/134
FETCH ComfyRegistry Data: 100/134
FETCH ComfyRegistry Data: 105/134
FETCH ComfyRegistry Data: 110/134
FETCH ComfyRegistry Data: 115/134
FETCH ComfyRegistry Data: 120/134
FETCH ComfyRegistry Data: 125/134
FETCH ComfyRegistry Data: 130/134
FETCH ComfyRegistry Data [DONE]
[ComfyUI-Manager] default cache updated: https://api.comfy.org/nodes
FETCH DATA from: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json [DONE]
[ComfyUI-Manager] All startup tasks have been completed.

Other

It's a workflow made from scratch but is the same as the ''Flux.2 [Klein] 9B Distilled: Image Edit'' from the templates tab, which is not working either. And yesterday worked

[GithubZeng]

comfyui-flux2fun-controlnet/flux_patch.py

[GithubZeng]

This file modified the initial forward_orig() function in comfy\ldm\flux\model.py
In Commit 44f1246, timestep_zero_index=None was added, while the custom node comfyui-flux2fun-controlnet didn't.
You can remove this node, or add timestep_zero_index=None in patched_forward_orig() function in comfyui-flux2fun-controlnet/flux_patch.py to fix this issue.

[GithubZeng]

Add timestep_zero_index=None, between line 74 and 75 in comfyui-flux2fun-controlnet/flux_patch.py like this:

def patched_forward_orig(
self,
img: Tensor,
img_ids: Tensor,
txt: Tensor,
txt_ids: Tensor,
timesteps: Tensor,
y: Tensor,
guidance: Tensor = None,
control=None,
timestep_zero_index=None,
transformer_options={},
attn_mask: Tensor = None,
) -> Tensor:

[jerocabezasgallego-bit]

Thanks! this solved the problem

[github-actions]

This issue is being marked stale because it has not had any activity for 30 days. Reply below within 7 days if your issue still isn't solved, and it will be left open. Otherwise, the issue will be closed automatically.