search

toriato / stable-diffusion-webui-wd14-tagger

Labeling extension for Automatic1111's Web UI
1.34k stars 235 forks source link

[Feature Request] Maximum number of tags #61

Open Evil-Dragon opened 1 year ago

Evil-Dragon commented 1 year ago

Would it be possible to have a variable limit set on the number of tags that the tagger can output to a file? For example you could set a maximum of 10 and only get those top 10 max tags for every file processed. This is particularly useful in dreambooth training. While I know threshold exists to limit the number of tags already it doesn't allow limiting for a maximum number of tags.

Many thanks.

picobyte commented 1 year ago

This is addressed in #93, in particular in commit 1196703.