mazesmazes commited on
Commit
50fd7ac
·
verified ·
1 Parent(s): f2359e2

Update custom model files, README, and requirements

Browse files
Files changed (2) hide show
  1. asr_config.py +2 -1
  2. asr_modeling.py +4 -0
asr_config.py CHANGED
@@ -45,7 +45,8 @@ class ASRConfig(transformers.PretrainedConfig):
45
  "max_new_tokens": 128,
46
  "min_new_tokens": 1,
47
  "do_sample": False,
48
- "repetition_penalty": 1.1,
 
49
  "no_repeat_ngram_size": 0,
50
  "use_cache": True,
51
  }
 
45
  "max_new_tokens": 128,
46
  "min_new_tokens": 1,
47
  "do_sample": False,
48
+ "repetition_penalty": 1.05,
49
+ "length_penalty": 0.9,
50
  "no_repeat_ngram_size": 0,
51
  "use_cache": True,
52
  }
asr_modeling.py CHANGED
@@ -206,6 +206,10 @@ class ASRModel(PreTrainedModel):
206
  self.decoder = self._create_decoder(config)
207
  self.generation_config = self.decoder.generation_config
208
 
 
 
 
 
209
  self._init_tokenizer()
210
 
211
  from types import SimpleNamespace
 
206
  self.decoder = self._create_decoder(config)
207
  self.generation_config = self.decoder.generation_config
208
 
209
+ # Set default generation parameters
210
+ self.generation_config.num_beams = 1
211
+ self.generation_config.length_penalty = 1.0
212
+
213
  self._init_tokenizer()
214
 
215
  from types import SimpleNamespace