stablediffusion吧 关注:30,954贴子:106,323
  • 17回复贴,共1

【求助】tagger拓展的model.onnx模型应该放在本地哪个文件夹下?

只看楼主收藏回复

这两天配置tagger的时候,因为没法直连huggingface,我就直接从抱脸下了它要的那个model.onnx,但是不知道它应该被放在哪个文件夹下。我看着model底下的这个文件夹挺像的(见图一),但是扔进去还是有图二那样的报错(看了md说明文件也没有明确说是哪个目录)
sd webui版本是v1.5.1,不过自带的两种图生文模型clip和deepbooru都没装,不知道会不会影响这个拓展的运行,有什么需要我补充的信息我都会尽快提供(帮帮焦头烂额的孩子(´;ω;`))




IP属地:陕西1楼2023-10-10 01:03回复
    没事了,干脆挂着小猫咪让它直接下了,最后居然是在一个这么诡异的路径里,这谁找得到。。。



    IP属地:陕西2楼2023-10-10 01:17
    回复
      想问下model.onnx是去哪下,方便给个链接吗


      IP属地:广东3楼2023-11-06 14:09
      收起回复
        给我和谐得找不着北了都。。。看图吧,有问题可以随时私信我


        IP属地:陕西4楼2023-11-06 20:22
        回复
          哎呀妈呀,感谢,同样问题!


          IP属地:辽宁6楼2023-11-20 18:38
          回复
            感谢万能的GPT大人,帮我解决了tagger第一次运行无法下载模型问题
            找到并打开这个文件:
            stable-diffusion-webui/extensions/stable-diffusion-webui-wd14-tagger/tagger/interrogator.py
            把download函数这部分代码改成下面这段:
            def download(self) -> Tuple[os.PathLike, os.PathLike]:
            print(f"Loading {self.name} model file from {self.kwargs['repo_id']}")
            model_path = Path(hf_hub_download(
            **self.kwargs, filename=self.model_path, endpoint="https://hf-mirror.com"))
            tags_path = Path(hf_hub_download(
            **self.kwargs, filename=self.tags_path, endpoint="https://hf-mirror.com"))
            return model_path, tags_path
            原理就是把huggingface.co指定成镜像站的地址hf-mirror.com,如果将来镜像站失效,再找别的镜像站就好。


            IP属地:辽宁7楼2023-11-21 10:58
            收起回复
              大佬大佬,我的用小猫咪无法下载,于是下载了model.onnx和csv文件,这两文件应该放在哪里啊


              IP属地:辽宁8楼2024-02-11 20:26
              回复
                大佬我的挂小猫咪也下不了这你知道啥情况吗


                IP属地:辽宁9楼2024-02-11 22:16
                收起回复
                  1. 直接将文件放在那个目录下不可行,它只会判断cache中是否有,没有的话还是得先下载。
                  2. 从https://hf-mirror.com 下载理论上可行,不过还是比较麻烦,最起码7楼大佬分享的代码已经不可用了。
                  最终我发现解决方法可以这样:
                  1. 从hf上下载好这俩文件,并存放在:
                  "model_path": "/stable-diffusion-webui/models/interrogators/models--SmilingWolf--wd-v1-4-moat-tagger-v2/snapshots/8452cddf280b952281b6e102411c50e981cb2908/model.onnx",
                  "tags_path": "/stable-diffusion-webui/models/interrogators/models--SmilingWolf--wd-v1-4-moat-tagger-v2/snapshots/8452cddf280b952281b6e102411c50e981cb2908/selected_tags.csv"
                  2. stable-diffusion-webui/extensions/stable-diffusion-webui-wd14-tagger/tagger/interrogator.py 代码中,先将下载标志位false,再将path直接修改为上述路径即可, 参考代码:
                  """ Interrogator for Waifu Diffusion models """
                  def __init__(
                  ....
                  is_hf=False,
                  ) -> None:
                  ....
                  def download(self) -> None:
                  mdir = Path(shared.models_path, 'interrogators')
                  if self.is_hf:
                  ......
                  else:
                  model_path = "/stable-diffusion-webui/models/interrogators/models--SmilingWolf--wd-v1-4-moat-tagger-v2/snapshots/8452cddf280b952281b6e102411c50e981cb2908/model.onnx"
                  tags_path = "/stable-diffusion-webui/models/interrogators/models--SmilingWolf--wd-v1-4-moat-tagger-v2/snapshots/8452cddf280b952281b6e102411c50e981cb2908/selected_tags.csv"


                  IP属地:上海10楼2024-03-25 15:48
                  收起回复
                    可以参考这个文档 https://blog.csdn.net/studio_1/article/details/142135456 试过了真可以


                    IP属地:越南12楼2024-11-18 11:01
                    回复