got an unexpected keyword argument 'use_cuda_graph'

[pyoliu]

Describe the bug

Cannot import /home/ubuntu/16T/lsm/ComfyUI/custom_nodes/ComfyUI-CogVideoXWrapper module for custom nodes: Failed to import diffusers.models.autoencoders.autoencoder_kl_cogvideox because of the following error (look up to see its traceback):
autotune() got an unexpected keyword argument 'use_cuda_graph'

Reproduction

python comfyui，Both version 0.31.0 and version 0.28.2 will experience the same issue

Logs

No response

System Info

cuda 12.2
torch 2.3.1
diffuers 0.31.0 / 0.28.2

Who can help?

No response

[a-r-r-o-w]

Gentle ping @kijai. Could you paste the entire stack trace? Currently there is no information where the autotune call or use_cuda_graph comes from

[kijai]

I have no idea, would need to see the whole stack indeed. Never seen such error before and I'm simply using
from diffusers.models import AutoencoderKLCogVideoX
in the code.

[pyoliu]

I have no idea, would need to see the whole stack indeed. Never seen such error before and I'm simply using in the code.from diffusers.models import AutoencoderKLCogVideoX

Traceback (most recent call last):
File "/home/ubuntu/16T/lsm/ComfyUI/nodes.py", line 1998, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in _call_with_frames_removed
File "/home/ubuntu/16T/lsm/ComfyUI/custom_nodes/ComfyUI-CogVideoXWrapper/init.py", line 1, in
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
File "/home/ubuntu/16T/lsm/ComfyUI/custom_nodes/ComfyUI-CogVideoXWrapper/nodes.py", line 8, in
from diffusers.models import AutoencoderKLCogVideoX, CogVideoXTransformer3DModel
File "", line 1075, in _handle_fromlist
File "/home/ubuntu/miniconda3/envs/sd3/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 843, in getattr
module = self._get_module(self._class_to_module[name])
File "/home/ubuntu/miniconda3/envs/sd3/lib/python3.10/site-packages/diffusers/utils/import_utils.py", line 855, in _get_module
raise RuntimeError(
RuntimeError: Failed to import diffusers.models.autoencoders.autoencoder_kl_cogvideox because of the following error (look up to see its traceback):
autotune() got an unexpected keyword argument 'use_cuda_graph'

Cannot import /home/ubuntu/16T/lsm/ComfyUI/custom_nodes/ComfyUI-CogVideoXWrapper module for custom nodes: Failed to import diffusers.models.autoencoders.autoencoder_kl_cogvideox because of the following error (look up to see its traceback):
autotune() got an unexpected keyword argument 'use_cuda_graph'

This is the specific information, and then I looked at more than one plugin reporting this error

[elsatch]

@pyoliu did you manage to solve it? I am getting a similar error with other plugin and I am not sure if a specific version of diffusers solves the problem.

Thanks in advance

[puke3615]

The same issue:

Cannot import ComfyUI/custom_nodes/ComfyUI-Easy-Use module for custom nodes: autotune() got an unexpected keyword argument 'use_cuda_graph'

And I found the source code in ComfyUI custom_nodes is:

from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block

My diffusers's version is 0.31.0.

[tin2tin]

Also got:
autotune() got an unexpected keyword argument 'use_cuda_graph'

But updating triton solved it.

[TIMESWALKING]

Also got: autotune() got an unexpected keyword argument 'use_cuda_graph'

But updating triton solved it.

me too