一、前言
最近想下载 stable diffusion 反推提示词插件 tagger ,也是好一番折腾,这里做个记录。
在安装之前确保能正常访问 github ,没有就安装个 steam++ ,如果会魔法就更好了,没有也不要紧。
二、下载
1、方式一
这种方式可以试试,可以成功自然最好,方便简单。但我试过了,不行,一直显示在下载,从后面下载成功的看来 tagger 体积也不大啊。
2、方式二
去 github 下载,下载地址为:
https://github.com/toriato/stable-diffusion-webui-wd14-tagger
直接到“根目录\extensions”,我的是:
D:\software\StableDiffusionWeb\stable-diffusion-webui\extensions
打开 git here ,执行如下命令:
git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git
等待下载完成
然后修改配置,先打开 stable-diffusion-webui-wd14-tagger 目录下的 preload.py 文件
做如下修改
# 原来的
# from modules.shared import models_path
# default_ddp_path = Path(models_path, 'deepdanbooru')
# 修改的
from modules import paths
default_ddp_path = Path(paths.models_path, 'deepdanbooru')
接着打开“根目录\extensions\stable-diffusion-webui-wd14-tagger\tagger”下的 ui.py 文件
做如下修改
# 原来的
# from webui import wrap_gradio_gpu_call
# 修改的
from modules.call_queue import wrap_gradio_gpu_call
tagger 需要从 huggingface 下载两个文件,会魔法的小伙伴看到这里就可以重启了,不会的可以继续往下看。
到这里下来那两个文件
新建一个目录,将下载好的两个文件放进去,我的目录是
D:\software\StableDiffusionWeb\stable-diffusion-webui\extensions\wd-v1-4-vit-tagger
接着“根目录\extensions\stable-diffusion-webui-wd14-tagger\tagger”下打开 interrogator.py 文件
在231行,将 model_path 改为 model.onnx 文件的绝对路径,tags_path 改为 selected_tags.csv 文件的绝对路径,如下图所示
然后重启
tagger 就可以使用了。