comfyui wd14 tagger 模型

<think>嗯，用户想要在ComfyUI中使用WD14 Tagger模型，或者下载适用于ComfyUI的WD14 Tagger模型。首先，我需要回忆一下WD14 Tagger的用途和安装步骤。根据提供的引用信息，这个扩展是用来从图像中提取booru标签的，可能用于自动标注图片。用户可能在使用过程中遇到了安装或配置的问题，或者想了解如何正确使用它。 首先，用户需要安装ComfyUI-WD14-Tagger扩展。根据引用中的项目地址，应该通过Git克隆仓库到ComfyUI的custom_nodes目录。然后安装依赖项，可能需要使用pip安装requirements.txt里的包。不过，用户可能遇到的常见问题比如CUDA或cuDNN的错误，引用[2]提到了ONNX运行时的CUDA加载错误，这可能与系统环境有关，需要确保正确安装了CUDA工具包和cuDNN，并且版本兼容。 接下来是模型下载部分。WD14 Tagger需要特定的模型文件，用户需要下载并放到正确的目录下。通常模型文件应该放在ComfyUI/models/taggers或者类似的路径下。用户可能不清楚具体路径，需要明确指出。此外，模型来源可能需要从Hugging Face下载，所以需要提供具体的下载链接或指导用户去官方仓库查找。 然后是配置和使用步骤。安装完成后，用户需要在ComfyUI中重新启动并加载节点，可能通过管理器来更新自定义节点列表。构建工作流时，可能需要使用特定的节点来加载图像和模型，并执行标记生成。用户可能需要了解每个节点的作用以及如何连接它们，例如图像输入、模型选择、输出标签等步骤。 常见问题部分需要考虑用户可能遇到的错误，比如缺少依赖项、模型路径错误、CUDA版本不匹配等。引用[2]中的错误提示说明用户需要检查CUDA和cuDNN的安装情况，或者尝试使用CPU模式作为备用方案。此外，模型文件是否正确下载和放置也是常见问题点，需要提醒用户验证文件路径和完整性。 用户可能的需求还包括性能优化，比如使用GPU加速，或者调整参数来提升标记生成的准确性。需要给出一些参数调整的建议，例如阈值设置、排除不必要的标签类别等。 最后，生成相关问题时要覆盖安装、配置、使用、错误处理以及高级应用等方面，帮助用户全面了解WD14 Tagger在ComfyUI中的使用。需要确保问题能够引导用户深入探索和解决可能遇到的问题。</think>以下是在ComfyUI中使用WD14 Tagger模型的完整指南： ### 一、安装步骤 1. **扩展安装** 通过Git克隆仓库到ComfyUI的`custom_nodes`目录： ```bash cd ComfyUI/custom_nodes git clone https://gitcode.com/gh_mirrors/ComfyUI-WD14-Tagger ``` 安装依赖库： ```bash pip install -r ComfyUI-WD14-Tagger/requirements.txt ``` 2. **模型下载** 需下载以下两个模型文件： - `wd14-vit.v310.safetensors`（标注模型） - `wd14-convnext.v310.safetensors`（图像编码模型） 从[Hugging Face仓库](https://huggingface.co/SmilingWolf/wd-v1-4-tagger)下载后，放置到： ``` ComfyUI/models/taggers/ ``` ### 二、使用流程 1. **节点加载** 重启ComfyUI后，在节点库中搜索`WD14Tagger`节点，包含： - `WD14TaggerLoader`（模型加载器） - `WD14Tagger`（标注执行器） 2. **工作流构建** ```python [图像输入节点] -> [WD14Tagger] ↗ [WD14TaggerLoader] ``` 需设置参数： - `threshold`：标签置信度阈值（默认0.35） - `character_threshold`：角色标签阈值（建议0.85） - `exclude_tags`：可屏蔽NSFW等特定标签 ### 三、常见问题解决 1. **CUDA加载错误** 若出现`ONNX Runtime CUDA error`： - 确认CUDA工具包版本≥11.8 - 更新cuDNN至8.9.4+ - 或改用CPU模式（修改代码中`providers=["CPUExecutionProvider"]`）[^2] 2. **模型未识别** 检查模型文件： ```bash ls ComfyUI/models/taggers/ # 应显示wd14-vit.v310.safetensors和wd14-convnext.v310.safetensors ``` ### 四、高级应用 通过`StringToText`节点将生成的标签接入文本提示系统，实现自动化图像标注流程。