ModuleNotFoundError: No module named ‘ai_diffusion’

快速结论：此报错通常出现在单独下载并执行 download_models.py 脚本时，而非通过完整的 krita-ai-diffusion 插件安装环境运行。优先排查是否克隆了整个插件仓库（包括子模块），而非仅下载单文件。

问题场景

用户自行手动执行 python download_models.py --all . 命令，试图为 ComfyUI 下载 AI Diffusion 模型，脚本位于 /media/HDD/programs/AI/ComfyUI/ 目录下。

报错原文

python download_models.py --all .
Traceback (most recent call last):
  File "/media/HDD/programs/AI/ComfyUI/download_models.py", line 26, in 
    from ai_diffusion import platform_tools
ModuleNotFoundError: No module named 'ai_diffusion'

原因分析

可能原因：download_models.py 脚本依赖于同仓库中的 ai_diffusion 模块（包含模型数据库和安装逻辑）。如果仅下载了 download_models.py 单个文件，或者仓库没有通过 git submodule update --init 初始化子模块，该模块将缺失，导致导入失败。

环境排查

确认当前 ComfyUI 目录下是否存在 custom_nodes/ 子目录。
确认是否通过 git clone 完整克隆了 Acly/krita-ai-diffusion 仓库，并执行了 git submodule update --init。
检查 ai_diffusion/ 目录是否存在于克隆下来的仓库根目录中。

解决步骤

进入 ComfyUI 的 custom_nodes 目录（若不存在则创建）：
cd /media/HDD/programs/AI/ComfyUI/custom_nodes
克隆 krita-ai-diffusion 仓库：
git clone https://github.com/Acly/krita-ai-diffusion.git
进入克隆后的目录并初始化子模块：
cd krita-ai-diffusion/
git submodule update --init
从仓库内部运行脚本，指定模型下载目标路径（根据你的设置调整）：
python scripts/download_models.py --all /media/HDD/programs/AI/ComfyUI

验证方法

执行完上述步骤后，脚本应能正常启动并开始下载模型。检查终端输出中是否有 Downloading model... 等字样，且无 ModuleNotFoundError 异常。

参考来源

Acly/krita-ai-diffusion #2521

AI 工具推荐

想把多个 AI 模型放在一个入口？

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型，适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接：通过此链接购买，我可能获得佣金，不影响你的价格。

ModuleNotFoundError: No module named ‘ai_diffusion’