【bug】可图文生图模型 KolorsPipeline IndexError: list index out of range

环境

linux
diffusers     0.30.0

问题详情

报错详情

from diffusers import KolorsPipelineTraceback (most recent call last):File "Kolors/demo.py", line 6, in <module>pipe = KolorsPipeline.from_pretrained(File ".conda/envs/xxx/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fnreturn fn(*args, **kwargs)File ".conda/envs/xxx/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 732, in from_pretrainedvariant_exists = is_folder and any(File ".conda/envs/xxx/lib/python3.10/site-packages/diffusers/pipelines/pipeline_utils.py", line 733, in <genexpr>p.split(".")[1].startswith(variant) for p in os.listdir(folder_path)
IndexError: list index out of range

运行代码

import torch
from diffusers import KolorsPipeline# 模型地址,可以改为本地离线地址
root_dir = "Kwai-Kolors/Kolors-diffusers"pipe = KolorsPipeline.from_pretrained(root_dir, torch_dtype=torch.float16, variant="fp16"
).to("cuda")
prompt = '一张瓢虫的照片，微距，变焦，高质量，电影，拿着一个牌子，写着"可图"'
image = pipe(prompt=prompt,negative_prompt="",guidance_scale=5.0,num_inference_steps=50,generator=torch.Generator(pipe.device).manual_seed(66),
).images[0]
# image.show()image.save("image.jpg")

错误说明

这个错误是由于模型中text_encoder文件夹下存在 __pycache__文件夹导致模型加载错误。

在这里插入图片描述

解决方法

解决方法是删除 text_encoder 文件夹下的 __pycache__文件夹。

我的模型是在可图 · 模型库 (modelscope.cn)下载的，官方没有去掉 __pycache__ 。

如果是在Kwai-Kolors/Kolors-diffusers at main (huggingface.co) 下载，官方去掉了 __pycache__文件夹，因此不会出现这个错误