03d7b39453
added an option to filter out deepbooru tags
3 years ago
60bd4d52a6
fix incorrect file extension filter for deepdanbooru models
3 years ago
563ea3f6ff
Change .cuda() to .to(devices.device)
3 years ago
c81d440d87
moved deepdanbooru to pure pytorch implementation
3 years ago
e38625011c
fix part2
3 years ago
72383abacd
Deepdanbooru linux fix
3 years ago
2e28c841f4
Oops
3 years ago
d62ef76614
Don't eat colons in booru tags
3 years ago
c24df4b486
Disable compiling deepbooru model
...
This is only necessary when you have to train,
and compiling model produces warning.
3 years ago
04c0e643f2
Merge branch 'master' of https://github.com/HunterVacui/stable-diffusion-webui
3 years ago
514456101b
[3/?] [wip] fix incorrect variable reference
...
still needs testing
3 years ago
f776254b12
[2/?] [wip] ignore OPT_INCLUDE_RANKS for training filenames
3 years ago
efefa4862c
[1/?] [wip] Reintroduce opts.interrogate_return_ranks
...
looks functionally correct, needs testing
Needs particular testing care around whether the colon usage (:) will break anything in whatever new use cases were introduced by https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/2143
3 years ago
698d303b04
deepbooru: added option to use spaces or underscores
...
deepbooru: added option to quote (\) in tags
deepbooru/BLIP: write caption to file instead of image filename
deepbooru/BLIP: now possible to use both for captions
deepbooru: process is stopped even if an exception occurs
3 years ago
f53f703aeb
resolved conflicts, moved settings under interrogate section, settings only show if deepbooru flag is enabled
3 years ago
ff4ef13dd5
removed unneeded print
3 years ago
76ef3d75f6
added deepbooru settings (threshold and sort by alpha or likelyhood)
3 years ago
b980e7188c
corrected tag return in get_deepbooru_tags
3 years ago
a1a05ad2d1
import time missing, added to deepbooru fixxing error on get_deepbooru_tags
3 years ago
1f92336be7
refactored the deepbooru module to improve speed on running multiple interogations in a row. Added the option to generate deepbooru tags for textual inversion preproccessing.
3 years ago
01f8cb4447
made deepdanbooru optional, added to readme, automatic download of deepbooru model
3 years ago
5f12e7efd9
linux test
3 years ago
fa2ea648db
even more powerfull fix
3 years ago
54fa613c83
loading tf only in interrogation process
3 years ago
4320f386d9
removing underscores and colons
3 years ago
17a99baf0c
better model search
3 years ago
1506fab29a
removing problematic tag
3 years ago
59a2b9e5af
deepdanbooru interrogator
3 years ago
AUTOMATIC
brkirch
Greendayle
guaneec
CookieHCl
Greg Fuller
JC_Array