[Bug] Node `NunchakuFluxDiTLoader` import failed:

[zwukong]

Checklist

• 1. I have searched for related issues and FAQs (https://github.com/mit-han-lab/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/mit-han-lab/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

Nunchaku version: 1.0.0.dev20250828
ComfyUI-nunchaku version: 1.0.0dev1
Node NunchakuFluxDiTLoader import failed:
Traceback (most recent call last):
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku_init_.py", line 32, in
from .nodes.models.flux import NunchakuFluxDiTLoader

Environment

3.12, 2.6

Reproduction Steps

启动后报错

[juntaosun]

After the update, all the previous Flux Kontext nodes ( NunchakuFluxDiTLoader ) disappeared?

[juntaosun]

ComfyUI-nunchaku 0.3.4 is not compatible with nunchaku 1.0.0.dev20250828.
Please update nunchaku to a supported version in ['v0.3.2'].

Node NunchakuFluxDiTLoader import failed.

Nunchaku's current version is a bit confusing.
The new version doesn't support the old version, Flux Kontext, and the old version doesn't support Qwen.

[lmxyy]

ComfyUI-nunchaku 0.3.4 is not compatible with nunchaku 1.0.0.dev20250828. Please update nunchaku to a supported version in ['v0.3.2'].

Node NunchakuFluxDiTLoader import failed.

Nunchaku's current version is a bit confusing. The new version doesn't support the old version, Flux Kontext, and the old version doesn't support Qwen.

Please upgrade your ComfyUI-nunchaku to the main branch.

[lmxyy]

Checklist

• 1. I have searched for related issues and FAQs (https://github.com/mit-han-lab/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/mit-han-lab/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

Nunchaku version: 1.0.0.dev20250828 ComfyUI-nunchaku version: 1.0.0dev1 Node NunchakuFluxDiTLoader import failed: Traceback (most recent call last): File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku__init__.py", line 32, in from .nodes.models.flux import NunchakuFluxDiTLoader

Environment

3.12, 2.6

Reproduction Steps

启动后报错

What is the detailed error?

[lmxyy]

@zwukong could you share your log?

[m8rr]

Nunchaku version: 1.0.0.dev20250828
ComfyUI-nunchaku version: 1.0.0dev1
Node NunchakuFluxDiTLoader import failed:
Traceback (most recent call last):
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes_init_.py", line 32, in
from .nodes.models.flux import NunchakuFluxDiTLoader
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\models\flux.py", line 22, in
from ...wrappers.flux import ComfyFluxWrapper
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in
from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)
Node NunchakuFluxLoraLoader import failed:
Traceback (most recent call last):
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes_init_.py", line 46, in
from .nodes.lora.flux import NunchakuFluxLoraLoader
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\lora\flux.py", line 14, in
from ...wrappers.flux import ComfyFluxWrapper
File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in
from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)

[lmxyy]

Nunchaku version: 1.0.0.dev20250828 ComfyUI-nunchaku version: 1.0.0dev1 Node NunchakuFluxDiTLoader import failed: Traceback (most recent call last): File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes__init__.py", line 32, in from .nodes.models.flux import NunchakuFluxDiTLoader File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\models\flux.py", line 22, in from ...wrappers.flux import ComfyFluxWrapper File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in from nunchaku.caching.utils import cache_context, create_cache_context ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py) Node NunchakuFluxLoraLoader import failed: Traceback (most recent call last): File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes__init__.py", line 46, in from .nodes.lora.flux import NunchakuFluxLoraLoader File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\lora\flux.py", line 14, in from ...wrappers.flux import ComfyFluxWrapper File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in from nunchaku.caching.utils import cache_context, create_cache_context ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)

In your case, either switch ComfyUI-nunchaku to the dev branch or use 0823 nightly wheels: https://github.com/nunchaku-tech/nunchaku/releases/tag/v1.0.0dev20250823

[Amazon90]

Nunchaku version: 1.0.0.dev20250828 ComfyUI-nunchaku version: 1.0.0dev1 Node NunchakuFluxDiTLoader import failed: Traceback (most recent call last): File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes__init__.py", line 32, in from .nodes.models.flux import NunchakuFluxDiTLoader File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\models\flux.py", line 22, in from ...wrappers.flux import ComfyFluxWrapper File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in from nunchaku.caching.utils import cache_context, create_cache_context ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py) Node NunchakuFluxLoraLoader import failed: Traceback (most recent call last): File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes__init__.py", line 46, in from .nodes.lora.flux import NunchakuFluxLoraLoader File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\nodes\lora\flux.py", line 14, in from ...wrappers.flux import ComfyFluxWrapper File "D:\AI\ComfyUI_windows_portable\ComfyUI\custom_nodes\nunchaku_nodes\wrappers\flux.py", line 15, in from nunchaku.caching.utils import cache_context, create_cache_context ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\AI\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)

In your case, either switch ComfyUI-nunchaku to the dev branch or use 0823 nightly wheels: https://github.com/nunchaku-tech/nunchaku/releases/tag/v1.0.0dev20250823

======================================== ComfyUI-nunchaku Initialization ========================================
Nunchaku version: 1.0.0.dev20250828
ComfyUI-nunchaku version: 1.0.0dev2
[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/model-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/extension-node-map.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json
Node `NunchakuFluxDiTLoader` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 32, in <module>
    from .nodes.models.flux import NunchakuFluxDiTLoader
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\flux.py", line 22, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
Node `NunchakuFluxLoraLoader` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 46, in <module>
    from .nodes.lora.flux import NunchakuFluxLoraLoader
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\lora\flux.py", line 14, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
D:\ComfyUI\venv\Lib\site-packages\timm\models\layers\__init__.py:48: FutureWarning: Importing from timm.models.layers is deprecated, please import via timm.layers
  warnings.warn(f"Importing from {__name__} is deprecated, please import via timm.layers", FutureWarning)
Nodes `NunchakuPulidApply`,`NunchakuPulidLoader`, `NunchakuPuLIDLoaderV2` and `NunchakuFluxPuLIDApplyV2` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 69, in <module>
    from .nodes.models.pulid import (
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\pulid.py", line 24, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
=================================================================================================================

[zwukong]

Nunchaku version: 1.0.0.dev20250828
ComfyUI-nunchaku version: 1.0.0dev1
Node NunchakuFluxDiTLoader import failed:
Traceback (most recent call last):
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku_init_.py", line 32, in
from .nodes.models.flux import NunchakuFluxDiTLoader
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\flux.py", line 22, in
from ...wrappers.flux import ComfyFluxWrapper
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in
from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (K:\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)
Node NunchakuFluxLoraLoader import failed:
Traceback (most recent call last):
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku_init_.py", line 46, in
from .nodes.lora.flux import NunchakuFluxLoraLoader
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\lora\flux.py", line 14, in
from ...wrappers.flux import ComfyFluxWrapper
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in
from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (K:\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)
FETCH ComfyRegistry Data: 10/95
Nodes NunchakuPulidApply,NunchakuPulidLoader, NunchakuPuLIDLoaderV2 and NunchakuFluxPuLIDApplyV2 import failed:
Traceback (most recent call last):
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku_init_.py", line 69, in
from .nodes.models.pulid import (
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\pulid.py", line 24, in
from ...wrappers.flux import ComfyFluxWrapper
File "K:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in
from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (K:\ComfyUI_windows_portable\python_embeded\Lib\site-packages\nunchaku\caching\utils.py)
相同的问题

[zwukong]

comfyui不能用manager切到dev分支，我看到你把nunchaku 0828release删了。只能回退了。暂时master与Nunchaku1.0.0dev20250823兼容，没报错

[Amazon90]

======================================== ComfyUI-nunchaku Initialization ========================================
Nunchaku version: 1.0.0.dev20250830
ComfyUI-nunchaku version: 1.0.0dev2
[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/model-list.json
Node `NunchakuFluxDiTLoader` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 32, in <module>
    from .nodes.models.flux import NunchakuFluxDiTLoader
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\flux.py", line 22, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
Node `NunchakuFluxLoraLoader` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 46, in <module>
    from .nodes.lora.flux import NunchakuFluxLoraLoader
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\lora\flux.py", line 14, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
[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/custom-node-list.json
[ComfyUI-Manager] default cache updated: https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/extension-node-map.json
Nodes `NunchakuPulidApply`,`NunchakuPulidLoader`, `NunchakuPuLIDLoaderV2` and `NunchakuFluxPuLIDApplyV2` import failed:
Traceback (most recent call last):
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\__init__.py", line 69, in <module>
    from .nodes.models.pulid import (
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\nodes\models\pulid.py", line 24, in <module>
    from ...wrappers.flux import ComfyFluxWrapper
  File "D:\ComfyUI\custom_nodes\ComfyUI-nunchaku\wrappers\flux.py", line 15, in <module>
    from nunchaku.caching.utils import cache_context, create_cache_context
ImportError: cannot import name 'cache_context' from 'nunchaku.caching.utils' (D:\ComfyUI\venv\Lib\site-packages\nunchaku\caching\utils.py)
=================================================================================================================

[Amazon90]

1.0.0.dev20250830 still no flux dit loader

[AiD-teng]

hi same issue here, so can i say the solution is " Either use 20250823 wheel OR switch to the branch? "

[zwukong]

Right, dev branch + latest wheel or master branch + 20250823 wheel