xiaolaa2 / stable-diffusion-webui-wd14-tagger

修改版,能够本地读取wd-tagger的模型,you can load wd-tagger model locally
16 stars 1 forks source link

怎么还是从网上加载啊? #1

Open crstudio opened 10 months ago

crstudio commented 10 months ago

Loading WD14 moat tagger v2 model file from SmilingWolf/wd-v1-4-moat-tagger-v2, model.onnx

我已经放在models下面了 image

crstudio commented 10 months ago

Loading WD14 moat tagger v2 model file from SmilingWolf/wd-v1-4-moat-tagger-v2, model.onnx

我已经放在models下面了 image

是扩展的目录必须改成 extensions/tagger 吗?

xiaolaa2 commented 9 months ago

不好意思,因为GitHub没有通知所以没及时看到你的issues。😢

我在服务器上试了最新版的webui以及在windows下试了B站up秋叶的整合包,都能够正常使用。

根据你的信息,你的wd-tagger加载的是wd-v1-4-moat-tagger-v2这个模型,但是目前我这个修改版的wd-tagger只有

安装方法和原版一样,就放到extensions文件夹里面 image

mingda27 commented 1 month ago

我试了好些时间才尝试成功的方法,成功的关键在于写对模型的名称。我用的是秋叶整合包。 把作者的这个下载后,解压到extensions文件夹,把原来的stable-diffusion-webui-wd14-tagger文件夹删掉。 编辑tagger下的utils.py文件,里面加上新模型,比如要用wd-v1-4-moat-tagger-v2,就在utils.py的第22行后面加上下面语句 'wd14-moat-v2': WaifuDiffusionInterrogator( 'wd-v1-4-moat-tagger-v2', repo_id='SmilingWolf/wd-v1-4-moat-tagger-v2', revision='v2.0', model_path=Path(wd_tagger_model_path, 'wd-v1-4-moat-tagger-v2','model.onnx'), tags_path=Path(wd_tagger_model_path, 'wd-v1-4-moat-tagger-v2','selected_tags.csv') ), 当然还要把模型文件model.onnx和selected_tags.csv,放在models\wd-tagger\wd-v1-4-moat-tagger-v2里面(作者在这个github的主页有说明)。重启后发现,拖入图片反推提示词,在控制台可以看到,加载了指定文件夹models\wd-tagger\下的模型,而不会自动下载到C盘的C:\Users\用户名\.cache\sdwebuilauncher\hfmirror\refs\SmilingWolf\里面去了。

########### 我把我用的utils.py文件文件放在这里了,解压后放到extensions\stable-diffusion-webui-wd14-tagger\tagger里面即可,缺少的模型去这里下载 https://huggingface.co/SmilingWolf 如果不想用某些模型,可以在utils.py文件里面注释掉。 utils.zip