Commit Graph

13 Commits

Author SHA1 Message Date
Guillaume Klein
e47e00910a Add length_penalty parameter and correctly compute the avg log prob 2023-02-22 10:27:38 +01:00
Guillaume Klein
f5c9f15c2c Check that the language code is valid 2023-02-21 12:10:54 +01:00
Guillaume Klein
e2094b6474 Reduce the maximum length when the prompt is longer than 448/2 2023-02-17 14:37:24 +01:00
Guillaume Klein
123d9a5704 Support English-only models 2023-02-16 17:02:40 +01:00
Guillaume Klein
cbbe633082 Add num_workers parameter 2023-02-14 09:34:05 +01:00
Guillaume Klein
c86353d323 Add task parameter 2023-02-13 21:26:25 +01:00
Guillaume Klein
f56dfc6491 Add without_timestamps parameter 2023-02-13 21:22:05 +01:00
Guillaume Klein
3dc44f7bb5 Raise a more explicit error message for English-only models 2023-02-13 18:26:45 +01:00
Guillaume Klein
269b3dfb10 Expose the device_index argument (#5) 2023-02-13 11:06:40 +01:00
Guillaume Klein
60e667e0d2 Cleanup unused import 2023-02-12 11:44:05 +01:00
Guillaume Klein
7d1d0541c8 Add the initial_prompt parameter (#2)
* Add the initial_prompt parameter

* Add docstring
2023-02-12 11:42:21 +01:00
Guillaume Klein
23d2d64259 Update transcribe.py 2023-02-11 11:47:07 +01:00
Guillaume Klein
5216d52d94 Initial commit 2023-02-11 10:21:19 +01:00