198 lines
10 KiB
JSON
198 lines
10 KiB
JSON
{
|
|
"name": "BooruTagger",
|
|
"logging": false,
|
|
"settings": {
|
|
"model": "wd-v1-4-moat-tagger-v2",
|
|
"threshold": 0.35,
|
|
"character_threshold": 0.85,
|
|
"exclude_tags": "",
|
|
"ortProviders": ["CUDAExecutionProvider", "CPUExecutionProvider"],
|
|
"preprocess": "tensor",
|
|
"ortIoBinding": true,
|
|
"ortIntraOpThreads": 0,
|
|
"HF_ENDPOINT": "https://huggingface.co"
|
|
},
|
|
"model_url": {
|
|
"wd-eva02-large-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-eva02-large-tagger-v3",
|
|
"wd-vit-large-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-vit-large-tagger-v3",
|
|
"wd-vit-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-vit-tagger-v3",
|
|
"wd-swinv2-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-swinv2-tagger-v3",
|
|
"wd-convnext-tagger-v3": "{HF_ENDPOINT}/SmilingWolf/wd-convnext-tagger-v3",
|
|
"camie-tagger-v2": "{HF_ENDPOINT}/Camais03/camie-tagger-v2",
|
|
"pixai-tagger-v0.9": "{HF_ENDPOINT}/deepghs/pixai-tagger-v0.9-onnx",
|
|
"pixai-tagger-v1.0-fp16": "{HF_ENDPOINT}/Mexes/pixai-tagger-v1.0-onnx-fp32-fp16-int8",
|
|
"pixai-tagger-v1.0-fp32": "{HF_ENDPOINT}/Mexes/pixai-tagger-v1.0-onnx-fp32-fp16-int8",
|
|
"cl-tagger-v2-v2_00": "{HF_ENDPOINT}/cella110n/cl_tagger_v2",
|
|
"cl-tagger-v2-v2_01a": "{HF_ENDPOINT}/cella110n/cl_tagger_v2",
|
|
"cl-tagger-v1-v1_00": "{HF_ENDPOINT}/cella110n/cl_tagger",
|
|
"cl-tagger-v1-v1_01": "{HF_ENDPOINT}/cella110n/cl_tagger",
|
|
"cl-tagger-v1-v1_02": "{HF_ENDPOINT}/cella110n/cl_tagger",
|
|
"wd-v1-4-moat-tagger-v2": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-moat-tagger-v2",
|
|
"wd-v1-4-convnextv2-tagger-v2": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-convnextv2-tagger-v2",
|
|
"wd-v1-4-convnext-tagger-v2": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-convnext-tagger-v2",
|
|
"wd-v1-4-vit-tagger-v2": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-vit-tagger-v2",
|
|
"wd-v1-4-swinv2-tagger-v2": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-swinv2-tagger-v2",
|
|
"wd-v1-4-convnext-tagger": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-convnext-tagger",
|
|
"wd-v1-4-vit-tagger": "{HF_ENDPOINT}/SmilingWolf/wd-v1-4-vit-tagger",
|
|
"animetimm-swinv2_base-dbv4-full": "{HF_ENDPOINT}/animetimm/swinv2_base_window8_256.dbv4-full",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "{HF_ENDPOINT}/animetimm/eva02_large_patch14_448.dbv4-full",
|
|
"animetimm-caformer_b36-dbv4-full": "{HF_ENDPOINT}/animetimm/caformer_b36.dbv4-full",
|
|
"animetimm-convnextv2_huge-dbv4-full": "{HF_ENDPOINT}/itterative/convnextv2_huge.dbv4-full-onnx"
|
|
},
|
|
"model_path": {
|
|
"wd-eva02-large-tagger-v3": "wd-eva02-large-tagger-v3/model.onnx",
|
|
"wd-vit-large-tagger-v3": "wd-vit-large-tagger-v3/model.onnx",
|
|
"wd-vit-tagger-v3": "wd-vit-tagger-v3/model.onnx",
|
|
"wd-swinv2-tagger-v3": "wd-swinv2-tagger-v3/model.onnx",
|
|
"wd-convnext-tagger-v3": "wd-convnext-tagger-v3/model.onnx",
|
|
"camie-tagger-v2": "camie-tagger-v2/camie-tagger-v2.onnx",
|
|
"pixai-tagger-v0.9": "pixai-tagger-v0.9/model.onnx",
|
|
"pixai-tagger-v1.0-fp16": "pixai-tagger-v1.0-fp16/model_fp16.onnx",
|
|
"pixai-tagger-v1.0-fp32": "pixai-tagger-v1.0-fp32/model.onnx",
|
|
"cl-tagger-v2-v2_00": "v2_00/model.onnx",
|
|
"cl-tagger-v2-v2_01a": "v2_01a/model.onnx",
|
|
"cl-tagger-v1-v1_00": "cl_tagger_1_00/model.onnx",
|
|
"cl-tagger-v1-v1_01": "cl_tagger_1_01/model.onnx",
|
|
"cl-tagger-v1-v1_02": "cl_tagger_1_02/model.onnx",
|
|
"wd-v1-4-moat-tagger-v2": "wd-v1-4-moat-tagger-v2/model.onnx",
|
|
"wd-v1-4-convnextv2-tagger-v2": "wd-v1-4-convnextv2-tagger-v2/model.onnx",
|
|
"wd-v1-4-convnext-tagger-v2": "wd-v1-4-convnext-tagger-v2/model.onnx",
|
|
"wd-v1-4-vit-tagger-v2": "wd-v1-4-vit-tagger-v2/model.onnx",
|
|
"wd-v1-4-swinv2-tagger-v2": "wd-v1-4-swinv2-tagger-v2/model.onnx",
|
|
"wd-v1-4-convnext-tagger": "wd-v1-4-convnext-tagger/model.onnx",
|
|
"wd-v1-4-vit-tagger": "wd-v1-4-vit-tagger/model.onnx",
|
|
"animetimm-swinv2_base-dbv4-full": "animetimm-swinv2_base-dbv4-full/model.onnx",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "animetimm-eva02_large_patch14_448-dbv4-full/model.onnx",
|
|
"animetimm-caformer_b36-dbv4-full": "animetimm-caformer_b36-dbv4-full/model.onnx",
|
|
"animetimm-convnextv2_huge-dbv4-full": "animetimm-convnextv2_huge-dbv4-full/model.onnx"
|
|
},
|
|
"metadata_path": {
|
|
"wd-eva02-large-tagger-v3": "wd-eva02-large-tagger-v3/selected_tags.csv",
|
|
"wd-vit-large-tagger-v3": "wd-vit-large-tagger-v3/selected_tags.csv",
|
|
"wd-vit-tagger-v3": "wd-vit-tagger-v3/selected_tags.csv",
|
|
"wd-swinv2-tagger-v3": "wd-swinv2-tagger-v3/selected_tags.csv",
|
|
"wd-convnext-tagger-v3": "wd-convnext-tagger-v3/selected_tags.csv",
|
|
"camie-tagger-v2": "camie-tagger-v2/camie-tagger-v2-metadata.json",
|
|
"pixai-tagger-v0.9": "pixai-tagger-v0.9/selected_tags.csv",
|
|
"pixai-tagger-v1.0-fp16": "pixai-tagger-v1.0-fp16/selected_tags.csv",
|
|
"pixai-tagger-v1.0-fp32": "pixai-tagger-v1.0-fp32/selected_tags.csv",
|
|
"cl-tagger-v2-v2_00": "v2_00/model_vocabulary.json",
|
|
"cl-tagger-v2-v2_01a": "v2_01a/model_vocabulary.json",
|
|
"cl-tagger-v1-v1_00": "cl_tagger_1_00/tag_mapping.json",
|
|
"cl-tagger-v1-v1_01": "cl_tagger_1_01/tag_mapping.json",
|
|
"cl-tagger-v1-v1_02": "cl_tagger_1_02/tag_mapping.json",
|
|
"wd-v1-4-moat-tagger-v2": "wd-v1-4-moat-tagger-v2/selected_tags.csv",
|
|
"wd-v1-4-convnextv2-tagger-v2": "wd-v1-4-convnextv2-tagger-v2/selected_tags.csv",
|
|
"wd-v1-4-convnext-tagger-v2": "wd-v1-4-convnext-tagger-v2/selected_tags.csv",
|
|
"wd-v1-4-vit-tagger-v2": "wd-v1-4-vit-tagger-v2/selected_tags.csv",
|
|
"wd-v1-4-swinv2-tagger-v2": "wd-v1-4-swinv2-tagger-v2/selected_tags.csv",
|
|
"wd-v1-4-convnext-tagger": "wd-v1-4-convnext-tagger/selected_tags.csv",
|
|
"wd-v1-4-vit-tagger": "wd-v1-4-vit-tagger/selected_tags.csv",
|
|
"animetimm-swinv2_base-dbv4-full": "animetimm-swinv2_base-dbv4-full/selected_tags.csv",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "animetimm-eva02_large_patch14_448-dbv4-full/selected_tags.csv",
|
|
"animetimm-caformer_b36-dbv4-full": "animetimm-caformer_b36-dbv4-full/selected_tags.csv",
|
|
"animetimm-convnextv2_huge-dbv4-full": "animetimm-convnextv2_huge-dbv4-full/selected_tags.csv"
|
|
},
|
|
"remote_model_path": {
|
|
"wd-eva02-large-tagger-v3": "model.onnx",
|
|
"wd-vit-large-tagger-v3": "model.onnx",
|
|
"wd-vit-tagger-v3": "model.onnx",
|
|
"wd-swinv2-tagger-v3": "model.onnx",
|
|
"wd-convnext-tagger-v3": "model.onnx",
|
|
"camie-tagger-v2": "camie-tagger-v2.onnx",
|
|
"pixai-tagger-v0.9": "model.onnx",
|
|
"pixai-tagger-v1.0-fp16": "model_fp16.onnx",
|
|
"pixai-tagger-v1.0-fp32": "model.onnx",
|
|
"cl-tagger-v2-v2_00": "v2_00/model.onnx",
|
|
"cl-tagger-v2-v2_01a": "v2_01a/model.onnx",
|
|
"cl-tagger-v1-v1_00": "cl_tagger_1_00/model.onnx",
|
|
"cl-tagger-v1-v1_01": "cl_tagger_1_01/model.onnx",
|
|
"cl-tagger-v1-v1_02": "cl_tagger_1_02/model.onnx",
|
|
"wd-v1-4-moat-tagger-v2": "model.onnx",
|
|
"wd-v1-4-convnextv2-tagger-v2": "model.onnx",
|
|
"wd-v1-4-convnext-tagger-v2": "model.onnx",
|
|
"wd-v1-4-vit-tagger-v2": "model.onnx",
|
|
"wd-v1-4-swinv2-tagger-v2": "model.onnx",
|
|
"wd-v1-4-convnext-tagger": "model.onnx",
|
|
"wd-v1-4-vit-tagger": "model.onnx",
|
|
"animetimm-swinv2_base-dbv4-full": "model.onnx",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "model.onnx",
|
|
"animetimm-caformer_b36-dbv4-full": "model.onnx",
|
|
"animetimm-convnextv2_huge-dbv4-full": "model.onnx"
|
|
},
|
|
"remote_metadata_path": {
|
|
"wd-eva02-large-tagger-v3": "selected_tags.csv",
|
|
"wd-vit-large-tagger-v3": "selected_tags.csv",
|
|
"wd-vit-tagger-v3": "selected_tags.csv",
|
|
"wd-swinv2-tagger-v3": "selected_tags.csv",
|
|
"wd-convnext-tagger-v3": "selected_tags.csv",
|
|
"camie-tagger-v2": "camie-tagger-v2-metadata.json",
|
|
"pixai-tagger-v0.9": "selected_tags.csv",
|
|
"pixai-tagger-v1.0-fp16": "selected_tags.csv",
|
|
"pixai-tagger-v1.0-fp32": "selected_tags.csv",
|
|
"cl-tagger-v2-v2_00": "v2_00/model_vocabulary.json",
|
|
"cl-tagger-v2-v2_01a": "v2_01a/model_vocabulary.json",
|
|
"cl-tagger-v1-v1_00": "cl_tagger_1_00/tag_mapping.json",
|
|
"cl-tagger-v1-v1_01": "cl_tagger_1_01/tag_mapping.json",
|
|
"cl-tagger-v1-v1_02": "cl_tagger_1_02/tag_mapping.json",
|
|
"wd-v1-4-moat-tagger-v2": "selected_tags.csv",
|
|
"wd-v1-4-convnextv2-tagger-v2": "selected_tags.csv",
|
|
"wd-v1-4-convnext-tagger-v2": "selected_tags.csv",
|
|
"wd-v1-4-vit-tagger-v2": "selected_tags.csv",
|
|
"wd-v1-4-swinv2-tagger-v2": "selected_tags.csv",
|
|
"wd-v1-4-convnext-tagger": "selected_tags.csv",
|
|
"wd-v1-4-vit-tagger": "selected_tags.csv",
|
|
"animetimm-swinv2_base-dbv4-full": "selected_tags.csv",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "selected_tags.csv",
|
|
"animetimm-caformer_b36-dbv4-full": "selected_tags.csv",
|
|
"animetimm-convnextv2_huge-dbv4-full": "selected_tags.csv"
|
|
},
|
|
"preprocess_path": {
|
|
"animetimm-swinv2_base-dbv4-full": "animetimm-swinv2_base-dbv4-full/preprocess.json",
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": "animetimm-eva02_large_patch14_448-dbv4-full/preprocess.json",
|
|
"animetimm-caformer_b36-dbv4-full": "animetimm-caformer_b36-dbv4-full/preprocess.json",
|
|
"animetimm-convnextv2_huge-dbv4-full": "animetimm-convnextv2_huge-dbv4-full/preprocess.json"
|
|
},
|
|
"preprocess_url": {
|
|
"animetimm-convnextv2_huge-dbv4-full": "{HF_ENDPOINT}/animetimm/convnextv2_huge.dbv4-full"
|
|
},
|
|
"external_data_path": {
|
|
"cl-tagger-v2-v2_00": "v2_00/model.onnx.data",
|
|
"cl-tagger-v2-v2_01a": "v2_01a/model.onnx.data",
|
|
"animetimm-convnextv2_huge-dbv4-full": "animetimm-convnextv2_huge-dbv4-full/model.onnx_data"
|
|
},
|
|
"remote_external_data_path": {
|
|
"animetimm-convnextv2_huge-dbv4-full": "model.onnx_data"
|
|
},
|
|
"threshold": {
|
|
"pixai-tagger-v0.9": 0.3,
|
|
"pixai-tagger-v1.0-fp16": 0.15,
|
|
"pixai-tagger-v1.0-fp32": 0.15,
|
|
"camie-tagger-v2": 0.492,
|
|
"cl-tagger-v1-v1_00": 0.55,
|
|
"cl-tagger-v1-v1_01": 0.55,
|
|
"cl-tagger-v1-v1_02": 0.55,
|
|
"cl-tagger-v2-v2_00": 0.55,
|
|
"cl-tagger-v2-v2_01a": 0.55,
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": 0.39,
|
|
"animetimm-caformer_b36-dbv4-full": 0.39,
|
|
"animetimm-swinv2_base-dbv4-full": 0.41,
|
|
"animetimm-convnextv2_huge-dbv4-full": 0.38
|
|
},
|
|
"character_threshold": {
|
|
"pixai-tagger-v0.9": 0.85,
|
|
"pixai-tagger-v1.0-fp16": 0.24,
|
|
"pixai-tagger-v1.0-fp32": 0.24,
|
|
"camie-tagger-v2": 0.492,
|
|
"cl-tagger-v1-v1_00": 0.6,
|
|
"cl-tagger-v1-v1_01": 0.6,
|
|
"cl-tagger-v1-v1_02": 0.6,
|
|
"cl-tagger-v2-v2_00": 0.55,
|
|
"cl-tagger-v2-v2_01a": 0.55,
|
|
"animetimm-eva02_large_patch14_448-dbv4-full": 0.61,
|
|
"animetimm-caformer_b36-dbv4-full": 0.47,
|
|
"animetimm-swinv2_base-dbv4-full": 0.59,
|
|
"animetimm-convnextv2_huge-dbv4-full": 0.51
|
|
}
|
|
}
|