mrprimenotes commited on
Commit
78e5ba3
·
verified ·
1 Parent(s): 6ae3ed8

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -4,6 +4,7 @@
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
6
  "auto_map": {
 
7
  "AutoModelForConditionalGeneration": "model.CustomWhisperForConditionalGeneration",
8
  "AutoModel": "model.CustomWhisperForConditionalGeneration",
9
  "AutoConfig": "model.CustomWhisperConfig"
 
4
  "activation_function": "gelu",
5
  "apply_spec_augment": false,
6
  "auto_map": {
7
+ "AutoModelForSpeechSeq2Seq": "model.CustomWhisperForConditionalGeneration",
8
  "AutoModelForConditionalGeneration": "model.CustomWhisperForConditionalGeneration",
9
  "AutoModel": "model.CustomWhisperForConditionalGeneration",
10
  "AutoConfig": "model.CustomWhisperConfig"