升级1.1后，运行到z-image dit 节点失败， #747
Description
Checklist
- 1. I have searched for related issues and FAQs (https://github.com/nunchaku-tech/nunchaku/blob/main/docs/faq.md) but was unable to find a solution.
- 2. The issue persists in the latest version.
- 3. Please note that without environment information and a minimal reproducible example, it will be difficult for us to reproduce and address the issue, which may delay our response.
- 4. If your report is a question rather than a bug, please submit it as a discussion at https://github.com/nunchaku-tech/ComfyUI-nunchaku/discussions/new/choose. Otherwise, this issue will be closed.
- 5. I will do my best to describe the issue in English.
Describe the Bug
ComfyUI Error Report
Error Details
- Node ID: 64
- Node Type: NunchakuZImageDiTLoader
- Exception Type: AssertionError
- Exception Message:
Stack Trace
File "D:\ComfyUI-aki-v2\ComfyUI\execution.py", line 516, 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:\ComfyUI-aki-v2\ComfyUI\execution.py", line 330, 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:\ComfyUI-aki-v2\ComfyUI\execution.py", line 304, in _async_map_node_over_list
await process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v2\ComfyUI\execution.py", line 292, in process_inputs
result = f(**inputs)
^^^^^^^^^^^
File "D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\zimage.py", line 216, in load_model
model = _load(sd, metadata=metadata)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\zimage.py", line 153, in _load
patch_scale_key(model.diffusion_model, patched_sd)
File "D:\ComfyUI-aki-v2\python\Lib\site-packages\nunchaku\models\transformers\utils.py", line 168, in patch_scale_key
assert state_dict[k].dtype == state_dict_from_checkpoint[k].dtype
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
System Information
- ComfyUI Version: 0.6.0
- Arguments: D:\ComfyUI-aki-v2\ComfyUI\main.py --auto-launch --preview-method auto --use-sage-attention --disable-cuda-malloc --fast
- OS: win32
- Python Version: 3.12.10 (tags/v3.12.10:0cc8128, Apr 8 2025, 12:21:36) [MSC v.1943 64 bit (AMD64)]
- Embedded Python: false
- PyTorch Version: 2.9.1+cu130
Devices
- Name: cuda:0 NVIDIA GeForce RTX 4070 Ti SUPER : cudaMallocAsync
- Type: cuda
- VRAM Total: 17170956288
- VRAM Free: 15779889152
- Torch VRAM Total: 33554432
- Torch VRAM Free: 25034752
Logs
2025-12-28T08:13:42.109908 - [START] Security scan2025-12-28T08:13:42.109908 -
2025-12-28T08:13:47.806584 - [DONE] Security scan2025-12-28T08:13:47.806584 -
2025-12-28T08:13:48.014642 - ## ComfyUI-Manager: installing dependencies done.2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** ComfyUI startup time:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - 2025-12-28 08:13:48.0142025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** Platform:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - Windows2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** Python version:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - 3.12.10 (tags/v3.12.10:0cc8128, Apr 8 2025, 12:21:36) [MSC v.1943 64 bit (AMD64)]2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** Python executable:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - D:\ComfyUI-aki-v2\python\python.exe2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** ComfyUI Path:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - D:\ComfyUI-aki-v2\ComfyUI2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** ComfyUI Base Folder Path:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - D:\ComfyUI-aki-v2\ComfyUI2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.014642 - ** User directory:2025-12-28T08:13:48.014642 - 2025-12-28T08:13:48.014642 - D:\ComfyUI-aki-v2\ComfyUI\user2025-12-28T08:13:48.014642 -
2025-12-28T08:13:48.015639 - ** ComfyUI-Manager config path:2025-12-28T08:13:48.015639 - 2025-12-28T08:13:48.015639 - D:\ComfyUI-aki-v2\ComfyUI\user\__manager\config.ini2025-12-28T08:13:48.015639 -
2025-12-28T08:13:48.015639 - ** Log path:2025-12-28T08:13:48.015639 - 2025-12-28T08:13:48.015639 - D:\ComfyUI-aki-v2\ComfyUI\user\comfyui.log2025-12-28T08:13:48.015639 -
2025-12-28T08:13:51.948713 -
Prestartup times for custom nodes:
2025-12-28T08:13:51.948713 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Easy-Use
2025-12-28T08:13:51.948713 - 10.1 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Manager
2025-12-28T08:13:51.948713 -
2025-12-28T08:13:54.975164 - Checkpoint files will always be loaded safely.
2025-12-28T08:13:55.126619 - Total VRAM 16376 MB, total RAM 32604 MB
2025-12-28T08:13:55.126619 - pytorch version: 2.9.1+cu130
2025-12-28T08:13:55.127617 - Enabled fp16 accumulation.
2025-12-28T08:13:55.127617 - Set vram state to: NORMAL_VRAM
2025-12-28T08:13:55.127617 - Device: cuda:0 NVIDIA GeForce RTX 4070 Ti SUPER : cudaMallocAsync
2025-12-28T08:13:55.138550 - Using async weight offloading with 2 streams
2025-12-28T08:13:55.139550 - Enabled pinned memory 14671.0
2025-12-28T08:13:55.147091 - working around nvidia conv3d memory bug.
2025-12-28T08:13:56.559212 - Using sage attention
2025-12-28T08:13:58.685665 - Python version: 3.12.10 (tags/v3.12.10:0cc8128, Apr 8 2025, 12:21:36) [MSC v.1943 64 bit (AMD64)]
2025-12-28T08:13:58.685665 - ComfyUI version: 0.6.0
2025-12-28T08:13:58.751064 - ComfyUI frontend version: 1.34.9
2025-12-28T08:13:58.753057 - [Prompt Server] web root: D:\ComfyUI-aki-v2\python\Lib\site-packages\comfyui_frontend_package\static
2025-12-28T08:13:58.778711 - D:\ComfyUI-aki-v2\python\Lib\site-packages\pydantic\_internal\_fields.py:151: UserWarning: Field "model_url" has conflict with protected namespace "model_".
You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.
warnings.warn(
2025-12-28T08:13:59.599998 - Total VRAM 16376 MB, total RAM 32604 MB
2025-12-28T08:13:59.599998 - pytorch version: 2.9.1+cu130
2025-12-28T08:13:59.599998 - Enabled fp16 accumulation.
2025-12-28T08:13:59.600990 - Set vram state to: NORMAL_VRAM
2025-12-28T08:13:59.600990 - Device: cuda:0 NVIDIA GeForce RTX 4070 Ti SUPER : cudaMallocAsync
2025-12-28T08:13:59.609964 - Using async weight offloading with 2 streams
2025-12-28T08:13:59.610957 - Enabled pinned memory 14671.0
2025-12-28T08:13:59.753646 - D:\ComfyUI-aki-v2\python\Lib\site-packages\pydantic\_internal\_fields.py:151: UserWarning: Field "model_name" has conflict with protected namespace "model_".
You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.
warnings.warn(
2025-12-28T08:13:59.797499 - D:\ComfyUI-aki-v2\python\Lib\site-packages\pydantic\_internal\_fields.py:151: UserWarning: Field "model_params" has conflict with protected namespace "model_".
You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.
warnings.warn(
2025-12-28T08:14:00.642826 - D:\ComfyUI-aki-v2\python\Lib\site-packages\pydantic\_internal\_fields.py:151: UserWarning: Field "model_version" has conflict with protected namespace "model_".
You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.
warnings.warn(
2025-12-28T08:14:00.643820 - D:\ComfyUI-aki-v2\python\Lib\site-packages\pydantic\_internal\_fields.py:151: UserWarning: Field "model_seed" has conflict with protected namespace "model_".
You may be able to resolve this warning by setting `model_config['protected_namespaces'] = ()`.
warnings.warn(
2025-12-28T08:14:01.258748 - [Crystools �[0;32mINFO�[0m] Crystools version: 1.27.4
2025-12-28T08:14:01.286721 - [Crystools �[0;32mINFO�[0m] Platform release: 10
2025-12-28T08:14:01.286721 - [Crystools �[0;32mINFO�[0m] JETSON: Not detected.
2025-12-28T08:14:01.287624 - [Crystools �[0;32mINFO�[0m] CPU: 12th Gen Intel(R) Core(TM) i5-12400F - Arch: AMD64 - OS: Windows 10
2025-12-28T08:14:01.297595 - [Crystools �[0;32mINFO�[0m] pynvml (NVIDIA) initialized.
2025-12-28T08:14:01.298588 - [Crystools �[0;32mINFO�[0m] GPU/s:
2025-12-28T08:14:01.310555 - [Crystools �[0;32mINFO�[0m] 0) NVIDIA GeForce RTX 4070 Ti SUPER
2025-12-28T08:14:01.310555 - [Crystools �[0;32mINFO�[0m] NVIDIA Driver: 591.44
2025-12-28T08:14:03.703332 - Exception in thread 2025-12-28T08:14:03.703332 - Thread-10 (_readerthread)2025-12-28T08:14:03.704333 - :
2025-12-28T08:14:03.704376 - Traceback (most recent call last):
2025-12-28T08:14:03.704376 - File "D:\ComfyUI-aki-v2\python\Lib\threading.py", line 1075, in _bootstrap_inner
2025-12-28T08:14:03.704376 - 2025-12-28T08:14:03.704882 - self.run()2025-12-28T08:14:03.704882 -
2025-12-28T08:14:03.704882 - File "<enhanced_experience vendors.sentry_sdk.integrations.threading>", line 99, in run
2025-12-28T08:14:03.704882 - File "<enhanced_experience vendors.sentry_sdk.integrations.threading>", line 94, in _run_old_run_func
2025-12-28T08:14:03.704882 - File "<enhanced_experience vendors.sentry_sdk.utils>", line 1649, in reraise
2025-12-28T08:14:03.704882 - File "<enhanced_experience vendors.sentry_sdk.integrations.threading>", line 92, in _run_old_run_func
2025-12-28T08:14:03.704882 - File "D:\ComfyUI-aki-v2\python\Lib\threading.py", line 1012, in run
2025-12-28T08:14:03.704882 - 2025-12-28T08:14:03.704882 - self._target(*self._args, **self._kwargs)2025-12-28T08:14:03.705879 -
2025-12-28T08:14:03.705879 - File "D:\ComfyUI-aki-v2\python\Lib\subprocess.py", line 1599, in _readerthread
2025-12-28T08:14:03.705879 - 2025-12-28T08:14:03.705879 - buffer.append(fh.read())2025-12-28T08:14:03.705879 -
2025-12-28T08:14:03.705879 - 2025-12-28T08:14:03.705879 - 2025-12-28T08:14:03.705879 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - 2025-12-28T08:14:03.706876 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 - ^2025-12-28T08:14:03.707872 -
2025-12-28T08:14:03.707872 - File "<frozen codecs>", line 322, in decode
2025-12-28T08:14:03.707872 - UnicodeDecodeError2025-12-28T08:14:03.707872 - : 2025-12-28T08:14:03.707872 - 'utf-8' codec can't decode byte 0xb2 in position 6: invalid start byte2025-12-28T08:14:03.707872 -
2025-12-28T08:14:06.095638 - [ComfyUI-Easy-Use] server: 2025-12-28T08:14:06.095638 - v1.3.4 2025-12-28T08:14:06.096635 - Loaded2025-12-28T08:14:06.096635 -
2025-12-28T08:14:06.096635 - [ComfyUI-Easy-Use] web root: 2025-12-28T08:14:06.096635 - D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Easy-Use\web_version/v2 2025-12-28T08:14:06.096635 - Loaded2025-12-28T08:14:06.096635 -
2025-12-28T08:14:06.104713 - ComfyUI-GGUF: Allowing full torch compile
2025-12-28T08:14:06.142972 - ### Loading: ComfyUI-Manager (V3.39)
2025-12-28T08:14:06.143970 - [ComfyUI-Manager] network_mode: public
2025-12-28T08:14:06.143970 - [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'.
2025-12-28T08:14:06.281056 - ### ComfyUI Version: v0.6.0 | Released on '2025-12-23'
2025-12-28T08:14:06.982083 - ======================================== ComfyUI-nunchaku Initialization ========================================
2025-12-28T08:14:06.984080 - Nunchaku version: 1.1.0
2025-12-28T08:14:06.985075 - ComfyUI-nunchaku version: 1.1.0
2025-12-28T08:14:07.006156 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/alter-list.json
2025-12-28T08:14:07.014332 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/model-list.json
2025-12-28T08:14:07.041522 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/github-stats.json
2025-12-28T08:14:07.096916 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/extension-node-map.json
2025-12-28T08:14:07.152729 - [ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json
2025-12-28T08:14:08.954602 - 'nunchaku_versions.json' not found. Node will start in minimal mode. Use 'update node' to fetch versions.2025-12-28T08:14:08.954602 -
2025-12-28T08:14:08.954602 - =================================================================================================================
2025-12-28T08:14:08.964915 - ⚡ SeedVR2 optimizations check: SageAttention ✅ | Flash Attention ❌ | Triton ✅2025-12-28T08:14:08.964915 -
2025-12-28T08:14:08.964915 - 💡 Optional: pip install flash-attn2025-12-28T08:14:08.964915 -
2025-12-28T08:14:08.964915 - 🔧 Conv3d workaround active: PyTorch 2.9.1, cuDNN 91200 (fixing VAE 3x memory bug)2025-12-28T08:14:08.964915 -
2025-12-28T08:14:09.098632 - 📊 Initial CUDA memory: 14.67GB free / 15.99GB total2025-12-28T08:14:09.098632 -
2025-12-28T08:14:10.100189 - [D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using ckpts path: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui_controlnet_aux\ckpts2025-12-28T08:14:10.100189 -
2025-12-28T08:14:10.100189 - [D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using symlinks: False2025-12-28T08:14:10.101188 -
2025-12-28T08:14:10.101188 - [D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui_controlnet_aux] | INFO -> Using ort providers: ['CUDAExecutionProvider', 'DirectMLExecutionProvider', 'OpenVINOExecutionProvider', 'ROCMExecutionProvider', 'MIGraphXExecutionProvider', 'CPUExecutionProvider', 'CoreMLExecutionProvider']2025-12-28T08:14:10.101188 -
2025-12-28T08:14:10.124172 - DWPose: Onnxruntime with acceleration providers detected2025-12-28T08:14:10.124172 -
2025-12-28T08:14:10.152006 -
Import times for custom nodes:
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\websocket_image_save.py
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-SCAIL-Pose
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-GGUF
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-WanAnimatePreprocess
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Custom-Scripts
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI_UltimateSDUpscale
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-segment-anything-2
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-KJNodes
2025-12-28T08:14:10.152006 - 0.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui_controlnet_aux
2025-12-28T08:14:10.152006 - 0.1 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\comfyui-tensorops
2025-12-28T08:14:10.152006 - 0.2 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-SeedVR2_VideoUpscaler
2025-12-28T08:14:10.152006 - 0.3 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-WanVideoWrapper
2025-12-28T08:14:10.152006 - 0.3 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-MelBandRoFormer
2025-12-28T08:14:10.152006 - 0.4 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-VideoHelperSuite
2025-12-28T08:14:10.152006 - 0.5 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Manager
2025-12-28T08:14:10.152006 - 0.6 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Crystools
2025-12-28T08:14:10.152006 - 2.0 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-nunchaku
2025-12-28T08:14:10.152006 - 4.8 seconds: D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-Easy-Use
2025-12-28T08:14:10.152006 -
2025-12-28T08:14:10.467104 - Context impl SQLiteImpl.
2025-12-28T08:14:10.467104 - Will assume non-transactional DDL.
2025-12-28T08:14:10.469100 - No target revision found.
2025-12-28T08:14:10.529895 - Starting server
2025-12-28T08:14:10.530891 - To see the GUI go to: http://127.0.0.1:8188
2025-12-28T08:14:11.095117 - [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.
2025-12-28T08:14:11.096112 - [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.
2025-12-28T08:14:11.100099 - [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.
2025-12-28T08:14:11.104140 - [DEPRECATION WARNING] Detected import of deprecated legacy API: /extensions/core/widgetInputs.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.
2025-12-28T08:14:11.685018 - FETCH ComfyRegistry Data: 5/1162025-12-28T08:14:11.685018 -
2025-12-28T08:14:11.821170 - [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.
2025-12-28T08:14:16.183329 - FETCH ComfyRegistry Data: 10/1162025-12-28T08:14:16.183329 -
2025-12-28T08:14:20.695792 - FETCH ComfyRegistry Data: 15/1162025-12-28T08:14:20.695792 -
2025-12-28T08:14:25.214313 - FETCH ComfyRegistry Data: 20/1162025-12-28T08:14:25.214313 -
2025-12-28T08:14:29.800214 - FETCH ComfyRegistry Data: 25/1162025-12-28T08:14:29.800214 -
2025-12-28T08:14:34.339473 - FETCH ComfyRegistry Data: 30/1162025-12-28T08:14:34.339473 -
2025-12-28T08:14:39.483377 - FETCH ComfyRegistry Data: 35/1162025-12-28T08:14:39.483377 -
2025-12-28T08:14:45.299021 - FETCH ComfyRegistry Data: 40/1162025-12-28T08:14:45.299021 -
2025-12-28T08:14:49.774391 - FETCH ComfyRegistry Data: 45/1162025-12-28T08:14:49.774391 -
2025-12-28T08:14:50.539293 - got prompt
2025-12-28T08:14:50.577613 - Using pytorch attention in VAE
2025-12-28T08:14:50.578610 - Using pytorch attention in VAE
2025-12-28T08:14:50.991825 - VAE load device: cuda:0, offload device: cpu, dtype: torch.bfloat16
2025-12-28T08:14:54.628942 - FETCH ComfyRegistry Data: 50/1162025-12-28T08:14:54.628942 -
2025-12-28T08:14:59.101912 - FETCH ComfyRegistry Data: 55/1162025-12-28T08:14:59.101912 -
2025-12-28T08:15:01.813496 - Requested to load ZImageTEModel_
2025-12-28T08:15:01.833815 - loaded completely; 95367431640625005117571072.00 MB usable, 7672.25 MB loaded, full load: True
2025-12-28T08:15:01.835808 - CLIP/text encoder model load device: cuda:0, offload device: cpu, current: cuda:0, dtype: torch.float16
2025-12-28T08:15:02.615194 - model weight dtype torch.float16, manual cast: None
2025-12-28T08:15:02.619123 - model_type FLOW
2025-12-28T08:15:02.730047 - !!! Exception during processing !!!
2025-12-28T08:15:02.731044 - Traceback (most recent call last):
File "D:\ComfyUI-aki-v2\ComfyUI\execution.py", line 516, 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:\ComfyUI-aki-v2\ComfyUI\execution.py", line 330, 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:\ComfyUI-aki-v2\ComfyUI\execution.py", line 304, in _async_map_node_over_list
await process_inputs(input_dict, i)
File "D:\ComfyUI-aki-v2\ComfyUI\execution.py", line 292, in process_inputs
result = f(**inputs)
^^^^^^^^^^^
File "D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\zimage.py", line 216, in load_model
model = _load(sd, metadata=metadata)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\ComfyUI-aki-v2\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\zimage.py", line 153, in _load
patch_scale_key(model.diffusion_model, patched_sd)
File "D:\ComfyUI-aki-v2\python\Lib\site-packages\nunchaku\models\transformers\utils.py", line 168, in patch_scale_key
assert state_dict[k].dtype == state_dict_from_checkpoint[k].dtype
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
AssertionError
2025-12-28T08:15:02.732041 - Prompt executed in 12.19 seconds
Attached Workflow
Please make sure that workflow does not contain any sensitive information such as API keys or passwords.
{"id":"9ae6082b-c7f4-433c-9971-7a8f65a3ea65","revision":0,"last_node_id":65,"last_link_id":81,"nodes":[{"id":9,"type":"SaveImage","pos":[1239.999958893878,259.99999997777456],"size":[780,660],"flags":{},"order":11,"mode":0,"inputs":[{"localized_name":"图片","name":"images","type":"IMAGE","link":45},{"localized_name":"文件名前缀","name":"filename_prefix","type":"STRING","widget":{"name":"filename_prefix"},"link":null}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.64","Node name for S&R":"SaveImage","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":["z-image"]},{"id":40,"type":"VAELoader","pos":[269.1333026991949,458.05755977116337],"size":[258.09895833333337,68.33333333333334],"flags":{"collapsed":true},"order":0,"mode":0,"inputs":[{"localized_name":"vae名称","name":"vae_name","type":"COMBO","widget":{"name":"vae_name"},"link":null}],"outputs":[{"localized_name":"VAE","name":"VAE","type":"VAE","links":[39]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.73","Node name for S&R":"VAELoader","models":[{"name":"ae.safetensors","url":"https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/vae/ae.safetensors","directory":"vae"}],"enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":["Z-image.safetensors"]},{"id":42,"type":"ConditioningZeroOut","pos":[132.4039464691183,556.2246414044986],"size":[140,26],"flags":{"collapsed":true},"order":7,"mode":0,"inputs":[{"localized_name":"条件","name":"conditioning","type":"CONDITIONING","link":36}],"outputs":[{"localized_name":"条件","name":"CONDITIONING","type":"CONDITIONING","links":[42]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.73","Node name for S&R":"ConditioningZeroOut","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":[]},{"id":43,"type":"VAEDecode","pos":[138.99270956832356,682.674003004168],"size":[258.99699777119446,46],"flags":{"collapsed":false},"order":10,"mode":0,"inputs":[{"localized_name":"Latent","name":"samples","type":"LATENT","link":38},{"localized_name":"vae","name":"vae","type":"VAE","link":39}],"outputs":[{"localized_name":"图像","name":"IMAGE","type":"IMAGE","slot_index":0,"links":[45]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.64","Node name for S&R":"VAEDecode","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":[]},{"id":45,"type":"CLIPTextEncode","pos":[442.6912424666129,313.90880231699487],"size":[435.95052083333337,355.9244791666667],"flags":{},"order":5,"mode":0,"inputs":[{"localized_name":"clip","name":"clip","type":"CLIP","link":54},{"localized_name":"文本","name":"text","type":"STRING","widget":{"name":"text"},"link":null}],"outputs":[{"localized_name":"条件","name":"CONDITIONING","type":"CONDITIONING","links":[36,41]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.73","Node name for S&R":"CLIPTextEncode","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":["冥府巨大城门的夜景，城门由黑色巨石与青铜铸成，两侧悬挂着成排发出幽绿色光芒的灯笼，灯火通明。一位身穿华丽紫色官袍、头戴乌纱、面容威严、蓄着长长白胡子的亚洲老者，手持卷宗，立于城门内侧。一位身穿半透明淡蓝色明代廪生服、面带安详微笑的中年灵魂，正轻盈地走向老者。城门周围缭绕着淡淡的雾气，远处可见影影绰绰的其他魂灵与鬼差。画面采用中景镜头，凸显人物与城门，冷色调为主，紫色与蓝色形成对比，氛围神秘而庄严。细节精细，8K分辨率。"],"color":"#232","bgcolor":"#353"},{"id":44,"type":"KSampler","pos":[900.8587512934861,275.0914978071577],"size":[326.53645833333337,635.8333333333334],"flags":{},"order":9,"mode":0,"inputs":[{"localized_name":"模型","name":"model","type":"MODEL","link":76},{"localized_name":"正面条件","name":"positive","type":"CONDITIONING","link":41},{"localized_name":"负面条件","name":"negative","type":"CONDITIONING","link":42},{"localized_name":"Latent图像","name":"latent_image","type":"LATENT","link":60},{"localized_name":"种子","name":"seed","type":"INT","widget":{"name":"seed"},"link":null},{"localized_name":"步数","name":"steps","type":"INT","widget":{"name":"steps"},"link":null},{"localized_name":"cfg","name":"cfg","type":"FLOAT","widget":{"name":"cfg"},"link":null},{"localized_name":"采样器名称","name":"sampler_name","type":"COMBO","widget":{"name":"sampler_name"},"link":null},{"localized_name":"调度器","name":"scheduler","type":"COMBO","widget":{"name":"scheduler"},"link":null},{"localized_name":"降噪","name":"denoise","type":"FLOAT","widget":{"name":"denoise"},"link":null}],"outputs":[{"localized_name":"Latent","name":"LATENT","type":"LATENT","slot_index":0,"links":[38]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.64","Node name for S&R":"KSampler","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":[712368560494503,"randomize",8,1,"res_multistep","simple",1]},{"id":48,"type":"LoraLoaderModelOnly","pos":[450.98438772268247,785.96851781495],"size":[210,99.84375],"flags":{},"order":8,"mode":4,"inputs":[{"localized_name":"模型","name":"model","type":"MODEL","link":75},{"localized_name":"LoRA名称","name":"lora_name","type":"COMBO","widget":{"name":"lora_name"},"link":null},{"localized_name":"模型强度","name":"strength_model","type":"FLOAT","widget":{"name":"strength_model"},"link":null}],"outputs":[{"localized_name":"模型","name":"MODEL","type":"MODEL","links":[76]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.75","Node name for S&R":"LoraLoaderModelOnly"},"widgets_values":["Z-image\\像素游戏_8b.safetensors",0.8]},{"id":51,"type":"CLIPLoader","pos":[145.00861446426876,461.065987615523],"size":[270,125],"flags":{"collapsed":true},"order":1,"mode":0,"inputs":[{"localized_name":"CLIP名称","name":"clip_name","type":"COMBO","widget":{"name":"clip_name"},"link":null},{"localized_name":"类型","name":"type","type":"COMBO","widget":{"name":"type"},"link":null},{"localized_name":"设备","name":"device","shape":7,"type":"COMBO","widget":{"name":"device"},"link":null}],"outputs":[{"localized_name":"CLIP","name":"CLIP","type":"CLIP","links":[54]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.76","Node name for S&R":"CLIPLoader"},"widgets_values":["qwen_3_4b.safetensors","lumina2","default"]},{"id":54,"type":"EmptyLatentImage","pos":[135.0746497121778,788.8657782841944],"size":[270.0772492408172,106],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"宽度","name":"width","type":"INT","widget":{"name":"width"},"link":null},{"localized_name":"高度","name":"height","type":"INT","widget":{"name":"height"},"link":null},{"localized_name":"批量大小","name":"batch_size","type":"INT","widget":{"name":"batch_size"},"link":null}],"outputs":[{"localized_name":"Latent","name":"LATENT","type":"LATENT","links":[60]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.56","Node name for S&R":"EmptyLatentImage"},"widgets_values":[1024,1024,1]},{"id":46,"type":"UNETLoader","pos":[131.572765297398,310.60933845538466],"size":[270,96.66666666666667],"flags":{},"order":3,"mode":4,"inputs":[{"localized_name":"UNet名称","name":"unet_name","type":"COMBO","widget":{"name":"unet_name"},"link":null},{"localized_name":"数据类型","name":"weight_dtype","type":"COMBO","widget":{"name":"weight_dtype"},"link":null}],"outputs":[{"localized_name":"模型","name":"MODEL","type":"MODEL","links":[]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.73","Node name for S&R":"UNETLoader","models":[{"name":"z_image_turbo_bf16.safetensors","url":"https://huggingface.co/Comfy-Org/z_image_turbo/resolve/main/split_files/diffusion_models/z_image_turbo_bf16.safetensors","directory":"diffusion_models"}],"enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":["z_image_turbo_bf16.safetensors","default"]},{"id":47,"type":"ModelSamplingAuraFlow","pos":[147.77007322546845,512.1300551996524],"size":[210,58],"flags":{"collapsed":false},"order":6,"mode":0,"inputs":[{"localized_name":"模型","name":"model","type":"MODEL","link":81},{"localized_name":"移位","name":"shift","type":"FLOAT","widget":{"name":"shift"},"link":null}],"outputs":[{"localized_name":"模型","name":"MODEL","type":"MODEL","slot_index":0,"links":[75]}],"properties":{"cnr_id":"comfy-core","ver":"0.3.64","Node name for S&R":"ModelSamplingAuraFlow","enableTabs":false,"tabWidth":65,"tabXOffset":10,"hasSecondTab":false,"secondTabText":"Send Back","secondTabOffset":80,"secondTabWidth":65},"widgets_values":[3]},{"id":64,"type":"NunchakuZImageDiTLoader","pos":[-172.42601209073004,328.8679710268276],"size":[271.6625269634046,58],"flags":{},"order":4,"mode":0,"inputs":[{"localized_name":"model_name","name":"model_name","type":"COMBO","widget":{"name":"model_name"},"link":null}],"outputs":[{"localized_name":"模型","name":"MODEL","type":"MODEL","links":[81]}],"properties":{"cnr_id":"ComfyUI-nunchaku","ver":"7bc610807d698a41a72a2cb125929cc7512d4959","Node name for S&R":"NunchakuZImageDiTLoader","aux_id":"nunchaku-tech/ComfyUI-nunchaku"},"widgets_values":["nunchaku\\int4-z-image-turbo.safetensors"]}],"links":[[36,45,0,42,0,"CONDITIONING"],[38,44,0,43,0,"LATENT"],[39,40,0,43,1,"VAE"],[41,45,0,44,1,"CONDITIONING"],[42,42,0,44,2,"CONDITIONING"],[45,43,0,9,0,"IMAGE"],[54,51,0,45,0,"CLIP"],[60,54,0,44,3,"LATENT"],[75,47,0,48,0,"MODEL"],[76,48,0,44,0,"MODEL"],[81,64,0,47,0,"MODEL"]],"groups":[{"id":2,"title":"Step2 - Image size","bounding":[434.46997384484655,703.7662606845104,451.90770854599265,206.58023894286828],"color":"#3f789e","font_size":24,"flags":{}},{"id":3,"title":"Step3 - Prompt","bounding":[430.26381018445164,235.25450504219478,459.162768828382,452.29921264262936],"color":"#3f789e","font_size":24,"flags":{}},{"id":4,"title":"Step1 - Load models","bounding":[120.2638101844517,235.25450504219478,298.4379689328665,345.6635321815364],"color":"#3f789e","font_size":24,"flags":{}},{"id":5,"title":"Group","bounding":[124.4885996045407,589.4507427127622,295.02063436920173,317.0203837424434],"color":"#3f789e","font_size":24,"flags":{}},{"id":6,"title":"双截棍","bounding":[-187.21733891368999,238.02566292829744,301.0029379844261,341.7933950966279],"color":"#3f789e","font_size":24,"flags":{}}],"config":{},"extra":{"ds":{"scale":0.6945101947704706,"offset":[337.48674769915453,123.1183256075004]},"frontendVersion":"1.32.9","workflowRendererVersion":"LG","VHS_latentpreview":false,"VHS_latentpreviewrate":0,"VHS_MetadataImage":true,"VHS_KeepIntermediate":true},"version":0.4}
Additional Context
(Please add any additional context or steps to reproduce the error here)
Environment
系统：w10
python：3.12
pytorch：2.9.1
cuda：130
Reproduction Steps
nunchakus官方z-image工作流，mangaer管理器更新