voiceSHIELD-small / config.json
sumitranjan's picture
Update config.json
e76a1d2 verified
raw
history blame contribute delete
505 Bytes
{
"architectures": ["VoiceShieldForAudioClassification"],
"auto_map": {
"AutoConfig": "modeling_voiceshield.VoiceShieldConfig",
"AutoModelForAudioClassification": "modeling_voiceshield.VoiceShieldForAudioClassification"
},
"base_model": "openai/whisper-small",
"model_type": "voiceshield",
"num_labels": 2,
"id2label": {
"0": "safe",
"1": "malicious"
},
"label2id": {
"safe": 0,
"malicious": 1
},
"torch_dtype": "float32",
"transformers_version": "4.36.0"
}