由于huggingface被墙,运行ComfyUI的WD14 Tagger插件无法自动下载模型。
本文介绍如何利用国内huggingface镜像来自助下载WD14 Tagger的模型。
下载地址
最常用的是Moat模型,可以在这里下载
https://hf-mirror.com/SmilingWolf/wd-v1-4-moat-tagger-v2/tree/main
包括onnx文件和csv文件

放置位置
下载后模型并不是放在ComfyUI的Models文件夹里,而是在WD14 Tagger的node文件夹里。
此外还要对下载下来的那个csv文件重命名,和onnx文件名一致。
正确文件结构
ComfyUI/custom_nodes/ComfyUI-WD14-Tagger/
└── models/
└── wd14_tagger/
├── wd-v1-4-moat-tagger-v2.onnx ← 模型文件
├── wd-v1-4-moat-tagger-v2.csv ← 重命名后的标签文件!
├── wd-v1-4-convnext-tagger-v2.onnx ← 其他模型
└── wd-v1-4-convnext-tagger-v2.csv ← 对应的标签文件操作步骤(以moat模型为例)
1. 下载文件
model.onnx → 重命名为 wd-v1-4-moat-tagger-v2.onnx
selected_tags.csv → 重命名为 wd-v1-4-moat-tagger-v2.csv
2. 放置路径
ComfyUI/custom_nodes/ComfyUI-WD14-Tagger/models/

PS看了源代码之后才知道要把csv文件和模型文件重命名成相同的名字:

