Update custom model files, README, and requirements
Browse files- asr_config.py +1 -1
asr_config.py
CHANGED
|
@@ -50,7 +50,7 @@ class ASRConfig(transformers.PretrainedConfig):
|
|
| 50 |
):
|
| 51 |
# Set default generation parameters (greedy decoding only)
|
| 52 |
generation_defaults = {
|
| 53 |
-
"num_beams":
|
| 54 |
"max_new_tokens": 256,
|
| 55 |
"repetition_penalty": 1.0,
|
| 56 |
"length_penalty": 1.0,
|
|
|
|
| 50 |
):
|
| 51 |
# Set default generation parameters (greedy decoding only)
|
| 52 |
generation_defaults = {
|
| 53 |
+
"num_beams": 1,
|
| 54 |
"max_new_tokens": 256,
|
| 55 |
"repetition_penalty": 1.0,
|
| 56 |
"length_penalty": 1.0,
|