wd-tagger-v3

[a-ru2016]

wd-taggerに新しいモデルが出たのでtag_images_by_wd14_tagger.pyで使えるようになりませんか？現在は以下のようなエラーが出ます。

/usr/local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  _torch_pytree._register_pytree_node(
/usr/local/lib/python3.10/site-packages/transformers/utils/generic.py:309: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  _torch_pytree._register_pytree_node(
/usr/local/lib/python3.10/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  torch.utils._pytree._register_pytree_node(
2024-03-14 05:03:15 INFO     downloading wd14    tag_images_by_wd14_tagger.py:86
                             tagger model from                                  
                             hf_hub. id:                                        
                             SmilingWolf/wd-swin                                
                             v2-tagger-v3                                       
/usr/local/lib/python3.10/site-packages/huggingface_hub/file_download.py:1146: FutureWarning: The `force_filename` parameter is deprecated as a new caching system, which keeps the filenames as they are on the Hub, is now in place.
  warnings.warn(
Traceback (most recent call last):
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_errors.py", line 286, in hf_raise_for_status
    response.raise_for_status()
  File "/usr/local/lib/python3.10/site-packages/requests/models.py", line 1021, in raise_for_status
    raise HTTPError(http_error_msg, response=self)
requests.exceptions.HTTPError: 404 Client Error: Not Found for url: https://huggingface.co/SmilingWolf/wd-swinv2-tagger-v3/resolve/main/keras_metadata.pb

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/tmp/content/sd-scripts/finetune/tag_images_by_wd14_tagger.py", line 386, in <module>
    main(args)
  File "/tmp/content/sd-scripts/finetune/tag_images_by_wd14_tagger.py", line 91, in main
    hf_hub_download(args.repo_id, file, cache_dir=args.model_dir, force_download=True, force_filename=file)
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
    return fn(*args, **kwargs)
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/file_download.py", line 1163, in hf_hub_download
    return cached_download(
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 118, in _inner_fn
    return fn(*args, **kwargs)
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/file_download.py", line 683, in cached_download
    r = _request_wrapper(
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/file_download.py", line 385, in _request_wrapper
    response = _request_wrapper(
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/file_download.py", line 409, in _request_wrapper
    hf_raise_for_status(response)
  File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_errors.py", line 296, in hf_raise_for_status
    raise EntryNotFoundError(message, response) from e
huggingface_hub.utils._errors.EntryNotFoundError: 404 Client Error. (Request ID: Root=1-65f28513-45cf78ce5111391b4a2a1574;3ed85aa7-15fb-4f01-8b82-9dcc101a201e)

Entry Not Found for url: https://huggingface.co/SmilingWolf/wd-swinv2-tagger-v3/resolve/main/keras_metadata.pb.
/usr/local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  _torch_pytree._register_pytree_node(
/usr/local/lib/python3.10/site-packages/transformers/utils/generic.py:309: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  _torch_pytree._register_pytree_node(
/usr/local/lib/python3.10/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
  torch.utils._pytree._register_pytree_node(
2024-03-14 05:03:22 INFO     found 12635 images. merge_dd_tags_to_metadata.py:18
                    INFO     loading existing    merge_dd_tags_to_metadata.py:24
                             metadata:                                          
                             /tmp/content/LoRA/m                                
                             eta_cap_dd.json    

[kohya-ss]

PR #1162 をいただいているので、時間でき次第、確認、マージいたします。