Emvo-ai
/

voiceSHIELD-small

Audio Classification

malicious-detection

Model card Files Files and versions

sumitranjan commited on Feb 17

Commit

e76a1d2

·

verified ·

1 Parent(s): 3051af2

Update config.json

Files changed (1) hide show

config.json +4 -2

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "architectures": ["VoiceShieldForAudioClassification"],
   "auto_map": {
     "AutoConfig": "modeling_voiceshield.VoiceShieldConfig",
-    "AutoModel": "modeling_voiceshield.VoiceShieldForAudioClassification"
   },
   "base_model": "openai/whisper-small",
   "model_type": "voiceshield",
@@ -14,5 +14,7 @@
   "label2id": {
     "safe": 0,
     "malicious": 1
-  }
 }

   "architectures": ["VoiceShieldForAudioClassification"],
   "auto_map": {
     "AutoConfig": "modeling_voiceshield.VoiceShieldConfig",
+    "AutoModelForAudioClassification": "modeling_voiceshield.VoiceShieldForAudioClassification"
   },
   "base_model": "openai/whisper-small",
   "model_type": "voiceshield",
   "label2id": {
     "safe": 0,
     "malicious": 1
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.36.0"
 }