luna
/
tagger-showdown
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update model ids

This commit is contained in:
Luna 2023-08-04 20:19:20 -03:00
parent 511ce6af16
commit 4b8202f493
3 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -40,7 +40,7 @@ cp config.example.json config.json
env/bin/python3 ./main.py fight env/bin/python3 ./main.py fight
# score mode -- crank the final numbers, generates graphs under plots/ folder # score mode -- crank the final numbers, generates graphs under plots/ folder
env/bin/python3 ./main.py fight env/bin/python3 ./main.py scores
# keep in mind that you can download more images, run fight mode, and then # keep in mind that you can download more images, run fight mode, and then
# run score mode! the commands are aware of work that's been already done and # run score mode! the commands are aware of work that's been already done and

16
main.py
View File

@ -24,13 +24,17 @@ from dataclasses import dataclass, field
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
# from https://github.com/picobyte/stable-diffusion-webui-wd14-tagger/blob/49cfdcd6be77086e6cc0d26b5ec26605b702be6e/tagger/utils.py#L15C1-L58C1
DEFAULTS = [ DEFAULTS = [
"wd14-vit-v2-git", "wd14-vit.v1",
"wd14-vit", "wd14-vit.v2",
"wd14-swinv2-v2-git", "wd14-convnext.v1",
"wd14-convnextv2-v2-git", "wd14-convnext.v2",
"wd14-convnext-v2-git", "wd14-convnextv2.v1",
"wd14-convnext", "wd14-swinv2-v1",
"wd-v1-4-moat-tagger.v2",
"mld-caformer.dec-5-97527",
# broken model: "mld-tresnetd.6-30000",
] ]

3
requirements.txt
View File

@ -1,8 +1,7 @@
aiosqlite==0.19.0 aiosqlite==0.19.0
aiohttp==3.8.4 aiohttp==3.8.4
aiolimiter>1.1.0<2.0 aiolimiter>1.1.0<2.0
aiofiles==23.1.0
plotly>5.15.0<6.0 plotly>5.15.0<6.0
pandas==2.0.2 pandas==2.0.2
kaleido==0.2.1 kaleido==0.2.1