mrprimenotes commited on
Commit
26ef198
·
verified ·
1 Parent(s): a2deb17

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -7,6 +7,7 @@
7
  "WhisperForConditionalGeneration"
8
  ],
9
  "auto_map": {
 
10
  "AutoModelForConditionalGeneration": "model.CustomWhisperForConditionalGeneration",
11
  "AutoModel": "model.CustomWhisperForConditionalGeneration",
12
  "AutoConfig": "model.CustomWhisperConfig"
 
7
  "WhisperForConditionalGeneration"
8
  ],
9
  "auto_map": {
10
+ "AutoModelForCTC": "model.CustomWhisperForConditionalGeneration",
11
  "AutoModelForConditionalGeneration": "model.CustomWhisperForConditionalGeneration",
12
  "AutoModel": "model.CustomWhisperForConditionalGeneration",
13
  "AutoConfig": "model.CustomWhisperConfig"