Captioning broken with wd-v1-4-convnexttv2-tagger-v2

快速结论：该报错发生在 Kohya SS 使用 ONNX 模式下载 wd-v1-4-convnextv2-tagger-v2 模型时，模型文件被下载到了错误路径或文件名不匹配，导致脚本在预期位置找不到 model.onnx。优先确认 sd-scripts 子模块是否已更新到包含修复的版本，或手动重命名/移动模型文件。

问题场景

用户使用 Kohya SS Windows GUI 版（Python 3.10），在 Captioning 界面启用了 ONNX 并勾选 “Force model download”，选择 SmilingWolf/wd-v1-4-convnextv2-tagger-v2 模型。模型文件虽被成功下载（日志显示 model.onnx: 100%），但脚本在 wd14_tagger_model\SmilingWolf_wd-v1-4-convnextv2-tagger-v2/model.onnx 路径下找不到文件，导致崩溃。

报错原文

loading onnx model: wd14_tagger_model\SmilingWolf_wd-v1-4-convnextv2-tagger-v2/model.onnx
Traceback (most recent call last):
  File "C:\Users\usr\Desktop\kohya_ss\sd-scripts\finetune\tag_images_by_wd14_tagger.py", line 515, in 
    main(args)
  File "C:\Users\usr\Desktop\kohya_ss\sd-scripts\finetune\tag_images_by_wd14_tagger.py", line 125, in main
    raise Exception(
Exception: onnx model not found: wd14_tagger_model\SmilingWolf_wd-v1-4-convnextv2-tagger-v2/model.onnx, please redownload the model with --force_download / onnxモデルが見つかりませんでした。--force_downloadで再ダウンロードしてください

原因分析

根本原因是模型下载的目录结构与脚本期望的路径不匹配。具体来说，脚本在 wd14_tagger_model/SmilingWolf_wd-v1-4-convnextv2-tagger-v2/model.onnx 查找文件，但下载工具将 model.onnx 放置在了不同的子目录或文件名下（例如可能带有 hash 后缀或放置在 models 子目录）。该问题与 #3178 同根同源，已在 sd-scripts 仓库的 PR #2053 中修复。

环境排查

确认 Kohya SS 版本：是否为最新 master 分支（包含 sd-scripts 子模块更新）。
检查 sd-scripts 子模块版本：确保已包含 PR #2053 的修复。
检查实际模型下载路径：查看 wd14_tagger_model/SmilingWolf_wd-v1-4-convnextv2-tagger-v2/ 目录下是否存在 model.onnx，以及是否有其他 .onnx 文件。
确认 ONNX runtime 已正确安装（虽然报错与 ONNX runtime 无关，但可一并排查）。

解决步骤

（可优先尝试）手动修复文件路径：进入 wd14_tagger_model/SmilingWolf_wd-v1-4-convnextv2-tagger-v2/ 目录，如果存在其他名称的 .onnx 文件（例如带有 hash 或版本号），将其重命名为 model.onnx，或移动到脚本期望的路径。
更新到最新版本：从 Kohya SS 的 master 分支重新拉取代码，确保 sd-scripts 子模块已更新到包含 PR #2053 修复的版本。
使用 –force_download 重新下载：在启用 ONNX 的情况下，重新勾选 “Force model download” 并运行 Captioning，让脚本重新下载完整的模型文件。注意：如果根因是路径映射问题，重新下载可能仍然失败。
临时替代方案：如果上述步骤均无效，可使用 A1111/reForge 扩展 wd14-tagger 作为临时替代，但需手动重命名和移动下载的文件。

验证方法

重新运行 Captioning 功能，观察日志中是否出现 loading onnx model: .../model.onnx 并正常加载，不再抛出 onnx model not found 异常。建议在一小批图片上测试，确认打标结果正确后，再全量运行。

参考来源

bmaltais/kohya_ss #3176

AI 工具推荐

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

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

了解 GamsGo AI

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

Captioning broken with wd-v1-4-convnexttv2-tagger-v2