Update custom model files, README, and requirements
Browse files- asr_config.py +3 -3
asr_config.py
CHANGED
|
@@ -45,12 +45,12 @@ class ASRConfig(transformers.PretrainedConfig):
|
|
| 45 |
# Set default generation parameters
|
| 46 |
generation_defaults = {
|
| 47 |
"num_beams": 1,
|
| 48 |
-
"max_new_tokens":
|
| 49 |
"min_new_tokens": 1,
|
| 50 |
"do_sample": False,
|
| 51 |
-
"repetition_penalty": 1.
|
| 52 |
"length_penalty": 1.0,
|
| 53 |
-
"no_repeat_ngram_size":
|
| 54 |
"use_cache": True,
|
| 55 |
}
|
| 56 |
|
|
|
|
| 45 |
# Set default generation parameters
|
| 46 |
generation_defaults = {
|
| 47 |
"num_beams": 1,
|
| 48 |
+
"max_new_tokens": 64,
|
| 49 |
"min_new_tokens": 1,
|
| 50 |
"do_sample": False,
|
| 51 |
+
"repetition_penalty": 1.0,
|
| 52 |
"length_penalty": 1.0,
|
| 53 |
+
"no_repeat_ngram_size": 0,
|
| 54 |
"use_cache": True,
|
| 55 |
}
|
| 56 |
|