diffusers/utils/import_utils.py No module named 'triton.ops'

[ZanePoe]

Describe the bug

` File "/home/zanepoe/miniconda3/envs/comfyui/lib/python3.12/site-packages/diffusers/utils/import_utils.py", line 910, in getattr
module = self._get_module(self._class_to_module[name])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/zanepoe/miniconda3/envs/comfyui/lib/python3.12/site-packages/diffusers/utils/import_utils.py", line 922, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.models.modeling_utils because of the following error (look up to see its traceback):
No module named 'triton.ops'

Cannot import /home/zanepoe/workspace/sd/ComfyUI/custom_nodes/ComfyUI-LTXVideo module for custom nodes: Failed to import diffusers.models.modeling_utils because of the following error (look up to see its traceback):
No module named 'triton.ops'
`

Reproduction

This error occurred when upgrading triton to 3.2.0

Logs

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/home/zanepoe/workspace/sd/ComfyUI/nodes.py", line 2147, in load_custom_node
    module_spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 999, in exec_module
  File "<frozen importlib._bootstrap>", line 488, in _call_with_frames_removed
  File "/home/zanepoe/workspace/sd/ComfyUI/custom_nodes/ComfyUI-MuseTalkUtils/__init__.py", line 4, in <module>
    from .musetalk_train_preprocess import *
  File "/home/zanepoe/workspace/sd/ComfyUI/custom_nodes/ComfyUI-MuseTalkUtils/musetalk_train_preprocess.py", line 8, in <module>
    from . import vae
  File "/home/zanepoe/workspace/sd/ComfyUI/custom_nodes/ComfyUI-MuseTalkUtils/vae.py", line 1, in <module>
    from diffusers import AutoencoderKL
  File "<frozen importlib._bootstrap>", line 1412, in _handle_fromlist
  File "/home/zanepoe/miniconda3/envs/comfyui/lib/python3.12/site-packages/diffusers/utils/import_utils.py", line 911, in __getattr__
    value = getattr(module, name)
            ^^^^^^^^^^^^^^^^^^^^^
  File "/home/zanepoe/miniconda3/envs/comfyui/lib/python3.12/site-packages/diffusers/utils/import_utils.py", line 910, in __getattr__
    module = self._get_module(self._class_to_module[name])
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/zanepoe/miniconda3/envs/comfyui/lib/python3.12/site-packages/diffusers/utils/import_utils.py", line 922, in _get_module
    raise RuntimeError(
RuntimeError: Failed to import diffusers.models.autoencoders.autoencoder_kl because of the following error (look up to see its traceback):
No module named 'triton.ops'

System Info

triton-3.2.0；
Name: torch：
Version: 2.6.0+cu126；
Name: diffusers：
Version: 0.32.2；
Python 3.12.8;
GPU:2080ti。

Who can help?

No response

[wenhao728]

TL;DR - I downgraded the torch to 2.5.1 and triton to 3.1.0 temporarily.

triton.ops was removed in Triton 3.1, but diffusers 0.32.2 still have lines of code depending on it.

Reference: woct0rdho/triton-windows#65 (comment)

It may take several days for huggingface team to integrate this. Before that, I downgraded triton and torch 2.6.0, which depends on triton 3.2.0.

[asomoza]

Hi! diffusers doesn't have any code that has triton as a dependency, going by the issue you linked it seems that it can be resolved by updating bitsandbytes. Can you test this?

You didn't post your bitsandbytes version, also I tested importing triton and diffusers with a clean install and plain vanilla diffusers and it doesn't produce any errors.

[wenhao728]

Hi! diffusers doesn't have any code that has triton as a dependency, going by the issue you linked it seems that it can be resolved by updating bitsandbytes. Can you test this?

You didn't post your bitsandbytes version, also I tested importing triton and diffusers with a clean install and plain vanilla diffusers and it doesn't produce any errors.

Thank you for your clarification. I didn't find any code involving import triton in diffusers.

I just tried to replicate the error, but I didn't get the previous errors as @ZanePoe .

Last time I got a similar error message and fixed it by downgrading triton. Maybe the error is from other packages. Adding CUDA_LAUNCH_BLOCKING=1 before python cli may help to locate the error more precisely.

[asomoza]

yeah, in ComfyUI people just install custom nodes with all the requirements each time so it can use a lot of libraries and do a lot of imports which can generate this kind of errors.

Using just diffusers with triton doesn't trigger any error so this is produced by a combination of libraries in OP environment (in this case from ComfyUI-MuseTalkUtils), probably the best way to deal with it is to just do a clean install.

[ZanePoe]

Hi! diffusers doesn't have any code that has triton as a dependency, going by the issue you linked it seems that it can be resolved by updating bitsandbytes. Can you test this?嗨! 扩散器没有任何代码,具有三元作为依赖项,通过您链接的问题,它似乎可以通过更新比特和字节来解决。 你能测试这个吗?

You didn't post your bitsandbytes version, also I tested importing triton and diffusers with a clean install and plain vanilla diffusers and it doesn't produce any errors.你没有发布你的比特字节版本,我也测试了导入三元和扩散器,带有干净的安装和普通香草扩散器,它不会产生任何错误。

Thank you, the problem was solved after I upgraded bitsandbytes

[asomoza]

glad it worked, I'll close this issue then