Commit Graph

39 Commits (65c738eee201cd38f5df1faf68efa7db638a751a)

Author SHA1 Message Date
Roel Kluin 7749166950 now all interrogators seem to work 2023-07-06 20:29:05 +02:00
Roel Kluin bc78d5ebe6 make cumulative work again 2023-07-06 20:08:04 +02:00
Roel Kluin 5514fb2517 fix for wd-v1-4-convnextv2-tagger-v2 2023-07-06 19:19:03 +02:00
Roel Kluin 40a19b0efd clickable tags 2023-07-06 16:17:51 +02:00
Roel Kluin 7b4907115c current 2023-07-06 10:00:06 +02:00
Roel Kluin eaf6cca1a1 add tab with excluded tags 2023-07-05 17:04:55 +02:00
Roel Kluin 818b86f166 warning from interrogation could not be unset 2023-07-05 12:12:44 +02:00
Roel Kluin a28d9d87e5 fix what linter indications 2023-07-05 10:41:05 +02:00
Roel Kluin eeae96e357 experimental large queries 2023-07-05 00:14:02 +02:00
Roel Kluin d99feaa93a bugfix 2023-06-22 19:25:58 +02:00
Roel Kluin 202a49377e 'fix' async issues by doing only simple actions on events 2023-06-22 01:31:31 +02:00
Roel Kluin eaf1fb0b5e isolate static parts from the interrogation 2023-06-22 00:26:09 +02:00
Roel Kluin 35cef34f72 bugfix 2023-06-20 19:44:13 +02:00
Roel Kluin f7c7cf5014 change add, exclude and search replace tags partially on lose focus 2023-06-20 19:37:55 +02:00
Roel Kluin 701581f6c5 fix 2023-06-20 18:26:19 +02:00
Roel Kluin 69fab9ef85 Add class for storage of queried data 2023-06-20 00:44:49 +02:00
Roel Kluin 0595035dd0 clean up 2023-06-19 22:06:48 +02:00
Roel Kluin 8321ea9fdb If there are multiple gpus this may be useful 2023-06-14 17:27:50 +02:00
Roel Kluin 4ee44d37c9 Fix: rereading the db.json should set the cls.ct past last number used.
The filters can be adapted later if the settings are erroneous,
but writing the tags files in that case is not right.
2023-06-13 23:17:45 +02:00
Roel Kluin 9ac2c216af better error messages 2023-06-13 19:29:03 +02:00
Roel Kluin f47d5da69c implement download strategy optimization suggested by Elziy 2023-06-13 17:04:38 +02:00
Roel Kluin 247cc986bb renaming 2023-06-13 15:45:47 +02:00
Roel Kluin 0b95d87181 moving code to interrogator 2023-06-13 15:18:44 +02:00
Roel Kluin e214f8de07 it is possible to set the threshold during the preprocessing 2023-06-07 22:25:34 +02:00
Sangha Lee 51a68b06f7 refactor: remove duplicated code 2023-01-30 18:51:16 +09:00
HalfMAI 493eba30a7 Update interrogator.py
Fix:
https://github.com/toriato/stable-diffusion-webui-wd14-tagger/issues/41
2023-01-30 17:33:17 +08:00
Sangha Lee 24be68585c fix: use tagged release rather than HEAD 2023-01-25 19:41:17 +09:00
SmilingWolf aca6753776 fix: preserve tags order
If a tag was changed, it was popped from the tags dict and added to the
bottom, changing their order.
2022-12-28 08:47:11 +01:00
Sangha Lee ce43eb74bd fix: commented out incorrectly 2022-12-26 20:34:10 +09:00
Sangha Lee dfb8d9d136 feat: unload models #31 2022-12-26 20:23:14 +09:00
SmilingWolf 953517f0b2
bugfix: properly check use_cpu
shared.cmd_opts.use_cpu is a list - compare accordingly
2022-12-10 22:12:19 +01:00
Sangha Lee ed5bc122c0 fix: set provider to onnx 2022-12-11 04:41:36 +09:00
Sangha Lee e22739d39c refactor: use tensorflow only when needed 2022-12-11 03:12:56 +09:00
Sangha Lee e4ec460122 feat: onnx model #22 2022-12-11 03:12:25 +09:00
Sangha Lee 6c2006f074 refactor: load model before evaluation 2022-12-11 02:28:55 +09:00
Sangha Lee c90ef1e815 refactor: replace double quotes 2022-12-10 15:05:43 +09:00
SmilingWolf ab7519bc13 Use up-to-date WD tagger model from HuggingFace repo
Also move model init to the first call to the interrogator instead of startup
2022-12-06 12:52:41 +01:00
Sangha Lee 2861f5e5b3 feat: addtional tags #18 2022-11-28 05:16:19 +09:00
Sangha Lee 047893b487 feat: support multiple model types #8 2022-11-21 23:36:25 +09:00