animeaidetect / config.json
hirawaru's picture
Upload config.json with huggingface_hub
da2a805 verified
Raw
History Blame Contribute Delete
506 Bytes
{
"architectures": ["EfficientNetForImageClassification"],
"model_type": "efficientnet",
"num_labels": 2,
"id2label": {
"0": "Natural",
"1": "Synthetic"
},
"label2id": {
"Natural": 0,
"Synthetic": 1
},
"custom_pipelines": {
"image-classification": {
"impl": "pipeline.AIImageDetectionPipeline",
"pt": ["AutoModel"]
}
},
"image_size": 224,
"dropout_rate": 0.3,
"torch_dtype": "float32",
"transformers_version": "4.35.0"
}