search

Particle1904 / DatasetHelpers

Dataset Helper program to automatically select, re scale and tag Datasets (composed of image and text) for Machine Learning training.
MIT License
157 stars 9 forks source link

SmilingWolf's eva02 tagger #34

Closed justTNP closed 1 month ago

justTNP commented 1 month ago

Adds SmilingWolf's eva02 large tagger, which has more training images and tags. Download links would have to be changed if merged, so I also made another pull request: https://huggingface.co/Crowlley/DatasetToolsModels/discussions/1.

EDIT: It's still on 2.8.0 and doesn't have any of the recent changes. I actually have no idea whether the repo I cloned today is on 2.8.0 or 2.8.3.

Particle1904 commented 1 month ago

Hey, thanks for adding support for the new model. I'll check the PR as soon as I can.

Particle1904 commented 1 month ago

Other than changing the WD Large input and output layer names, everything is working. Again, thanks for the contribution.