ComfyUI-BrushNet项目模型加载问题解析
项目地址: https://gitcode.com/gh_mirrors/co/ComfyUI-BrushNet 在使用ComfyUI-BrushNet项目时，用户可能会遇到"'BrushNetModel' object has no attribute 'add_embedding'"的错误提示。这个问题通常是由于模型文件类型不匹配导致的。
问题本质
该错误的核心在于用户尝试加载了不兼容的模型文件格式。BrushNet项目针对不同版本的Stable Diffusion模型提供了专门的实现：
- 针对SD1.5版本的BrushNet模型使用.safetensors格式
- 针对SDXL版本的BrushNet模型使用.ckpt格式
当用户尝试使用SDXL的.ckpt格式模型加载到SD1.5版本的BrushNet实现中时，就会出现上述属性缺失的错误。
解决方案
要解决这个问题，用户需要确保：
- 确认自己使用的是SD1.5还是SDXL版本的BrushNet实现
- 下载对应版本的模型文件
- 将模型文件放置在正确的目录路径下
对于SD1.5版本，必须使用.safetensors格式的模型文件；而对于SDXL版本，则应使用.ckpt格式的模型文件。
技术背景
这个错误反映了深度学习模型加载过程中的一个常见问题：模型架构与权重文件不匹配。BrushNetModel类在设计时针对不同版本的Stable Diffusion实现了不同的接口方法，其中add_embedding方法是SD1.5版本特有的功能。
当加载器尝试加载模型权重时，如果发现权重文件与当前模型架构不兼容，就会抛出属性缺失的错误。这种设计有助于开发者快速识别模型版本不匹配的问题，而不是在后续推理过程中出现更隐蔽的错误。
最佳实践
为了避免类似问题，建议用户：
- 仔细阅读项目文档，了解不同版本模型的要求
- 在下载模型文件时确认其对应的Stable Diffusion版本
- 在加载模型前检查文件扩展名是否符合预期
- 保持ComfyUI和相关依赖库的最新版本
通过遵循这些实践，可以显著减少模型加载过程中出现的问题，提高工作效率。
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
