Update custom model files, README, and requirements
Browse files- asr_config.py +2 -1
- 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.
|
|
|
|
| 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
|