-
-
Notifications
You must be signed in to change notification settings - Fork 7
Add compatibility with timm-based models and more #52
Copy link
Copy link
Open
Labels
ElectronRequires changes in the electron sideRequires changes in the electron sideFeatureNew feature or requestNew feature or requestFrontendRequires changes in the UIRequires changes in the UIPythonRequires changes in the python scriptsRequires changes in the python scripts
Description
Activity
Metadata
Metadata
Assignees
Labels
ElectronRequires changes in the electron sideRequires changes in the electron sideFeatureNew feature or requestNew feature or requestFrontendRequires changes in the UIRequires changes in the UIPythonRequires changes in the python scriptsRequires changes in the python scripts
Describe the feature idea
Add compatibility with timm-based models. (ie. https://huggingface.co/animetimm).
Add compatibility with safentensors.
Add an option that follows the calibrated threshold available in the selected_tags.csv.
Add a field to add thresholds.csv when provided by the model available.
Add an option that follows both the calibrated thresholds in "selected_tags" and whats in "thresholds".
Some models only have an artist category, the tagger ignores that.
Tagger also ignores rating argmax.