ImportError: cannot import name 'CogView4LoraLoaderMixin' from 'diffusers.loaders'

[godwenbin]

Describe the bug

Traceback (most recent call last):
File "/root/mambaforge/envs/gwb_kohya/bin/cogkit", line 5, in
from cogkit.cli import cli
File "/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/cogkit/init.py", line 4, in
from cogkit.python import generate_image, generate_video
File "/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/cogkit/python/init.py", line 4, in
from .generation.image import generate_image
File "/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/cogkit/python/generation/image.py", line 11, in
from cogkit.utils import (
File "/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/cogkit/utils/init.py", line 6, in
from cogkit.utils.lora import load_lora_checkpoint, unload_lora_checkpoint
File "/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/cogkit/utils/lora.py", line 4, in
from diffusers.loaders import CogVideoXLoraLoaderMixin, CogView4LoraLoaderMixin
ImportError: cannot import name 'CogView4LoraLoaderMixin' from 'diffusers.loaders' (/root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/diffusers/loaders/init.py)

Reproduction

1

Logs

System Info

diffusers 0.33.0.dev0
linux
python 3.10.14

Who can help?

No response

[yiyixuxu]

hi @godwenbin
thank for reporting the issue
i cannot reproduce from my end, can you run this directly with diffusers?

from diffusers.loaders import CogVideoXLoraLoaderMixin, CogView4LoraLoaderMixin

[hlky]

Hi @godwenbin. It looks like you are reusing a venv from kohya, gwb_kohya, when installing cogkit it's possible the venv didn't update diffusers as it was already installed. Can we double check the installed version, look for __version__ in /root/mambaforge/envs/gwb_kohya/lib/python3.10/site-packages/diffusers/__init__.py. If possible, try cogkit in a fresh venv, or reinstall diffusers in the existing venv with pip install -U git+https://github.com/huggingface/diffusers.git