Skip to content
This repository has been archived by the owner on Jul 17, 2023. It is now read-only.

Optimization issue + wd-v1-4-moat-tagger-v2 #98

Open
majen-dev opened this issue Jun 21, 2023 · 1 comment
Open

Optimization issue + wd-v1-4-moat-tagger-v2 #98

majen-dev opened this issue Jun 21, 2023 · 1 comment

Comments

@majen-dev
Copy link

Hi, I would like to report a few things.

  1. The performance is really poor, it uses GPU but the GPU is sitting at 0% most of the time and CPU is doing some 70% heavy lifting. Even if it's standardising images on CPU first no way it's that slow. It does like 1 image/sec that's terrible if you want to do 150k.
  2. Also wd-v1-4-moat-tagger-v2 is currently best or one of best WD taggers but it's not supported
@wensleyoliv
Copy link

I added the following lines in the utils.py file in the extension folder and i'm using moat this way.

'wd14-MOAT-v2': WaifuDiffusionInterrogator(
'wd14-MOAT-v2',
repo_id='SmilingWolf/wd-v1-4-moat-tagger-v2'

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants