卓特视觉
AI生成视频

LCM_Inpaint_Outpaint_Comfy导入失败,提示cannot import name ‘CrossAttnDownBlock2D’ from ‘diffusers.models.unets.unet_2d_condition’的解决办法

之前我已经写过一篇文章来解决No module named ‘diffusers.models.unet_2d_blocks’的问题,具体参考如下

 

但是,随着diffusers的更新,在使用插件LCM_Inpaint_Outpaint_Comfy时,又遇到了新的问题,找不到 CrossAttnDownBlock2D 和 CrossAttnUpBlock2D,导入失败,日志报错:

from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel,CrossAttnDownBlock2D, CrossAttnUpBlock2D, DownBlock2D, UpBlock2D
ImportError: cannot import name 'CrossAttnDownBlock2D' from 'diffusers.models.unets.unet_2d_condition'
LCM_Inpaint_Outpaint_Comfy导入失败,提示cannot import name 'CrossAttnDownBlock2D' from 'diffusers.models.unets.unet_2d_condition'的解决办法

这个问题的解决方式跟之前类似,原因也大致相同,只需将custom_nodes\LCM_Inpaint_Outpaint_Comfy\LCM\LCM_reference_pipeline.py文件中下面的代码,大致在37行

from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel,CrossAttnDownBlock2D, CrossAttnUpBlock2D, DownBlock2D, UpBlock2D

修改为

from diffusers.models.unets.unet_2d_blocks import CrossAttnDownBlock2D, CrossAttnUpBlock2D, DownBlock2D, UpBlock2D
from diffusers.models.unets.unet_2d_condition import UNet2DConditionModel

修改完成后保存,然后重启comfyui,即可正常导入LCM_Inpaint_Outpaint_Comfy!

© 版权声明

相关文章