Commit
•
bdac110
1
Parent(s):
18530d7
Add `return_timestamps` attribute to generation config
Browse filesFor consistency with the other Whisper models, we should set the `return_timestamps` attribute to `False` in the generation config, e.g.: for Whisper Small: https://huggingface.co/openai/whisper-small/blob/e34e8ae444c29815eca53e11383ea13b2e362eb0/generation_config.json#L167
cc
@patrickvonplaten
if you could merge that would be grand! Thanks!
- generation_config.json +1 -0
generation_config.json
CHANGED
@@ -148,6 +148,7 @@
|
|
148 |
"max_length": 448,
|
149 |
"no_timestamps_token_id": 50363,
|
150 |
"pad_token_id": 50257,
|
|
|
151 |
"suppress_tokens": [
|
152 |
1,
|
153 |
2,
|
|
|
148 |
"max_length": 448,
|
149 |
"no_timestamps_token_id": 50363,
|
150 |
"pad_token_id": 50257,
|
151 |
+
"return_timestamps": false,
|
152 |
"suppress_tokens": [
|
153 |
1,
|
154 |
2,
|