Jong Wook Kim
68e44bd83c
saving the qk matrix in the attention module for convenience
2022-12-29 23:02:52 -07:00
Jong Wook Kim
0b5dcfdef7
large-v2 figure and arxiv url update
2022-12-09 00:12:39 -05:00
Jong Wook Kim
4179ed2475
add large-v2 model
...
- The "large-v2" model is trained for more epochs with regularization and shows improved performance compared to the previous large.
- It has the same architecture as the original large model.
- When `load_model("large")` is called, the "large-v2" model will be loaded.
- We will soon update the paper regarding this new model.
2022-12-05 11:07:14 -05:00
Jong Wook Kim
eff383b27b
invoking __call__ instead of forward()
2022-11-16 04:18:50 -08:00
Jong Wook Kim
02aa851a49
fix to return only the text token ids
2022-11-15 16:25:11 -08:00
Jong Wook Kim
d18e9ea5dd
transcribe() on English-only model won't complain when language="en" is not given
2022-10-09 02:40:12 -07:00
Jong Wook Kim
7cb4cc21bf
allowing nonzero initial temperature
2022-09-29 18:05:12 -07:00
Jong Wook Kim
30dc5c581b
pointer to the show and tell section
2022-09-29 14:57:49 -07:00
Jong Wook Kim
62fe7f1009
patience definition to match the paper
2022-09-27 19:00:41 -07:00
Jong Wook Kim
5d8d3e75a4
add --condition_on_previous_text
2022-09-25 05:16:08 -07:00
Jong Wook Kim
2d3032de01
improved warning message for English-only models
2022-09-25 02:10:36 -07:00
Jong Wook Kim
8cf36f3508
allow hyphens and single quotes between words
2022-09-23 20:11:27 +09:00
Jong Wook Kim
15ab548263
nocaptions -> nospeech to match the paper figure
2022-09-23 15:45:32 +09:00
Jong Wook Kim and GitHub
e90b8fa7e8
Merge pull request #14 from bquast/patch-1
...
make LICENSE a link instead of code-formatted text
2022-09-22 11:51:05 +09:00
Jong Wook Kim and GitHub
f83cb83a42
Merge pull request #24 from ldanilov/patch-1
...
fixes the link to the model paper
2022-09-22 11:48:57 +09:00
Jong Wook Kim
49a3ffc997
add section Available models and languages
2022-09-22 05:36:25 +09:00
Jong Wook Kim
cfd6bdda21
a note on speed-accuracy tradeoffs
2022-09-22 02:58:56 +09:00
Jong Wook Kim
834f00a0ea
making small model the default
2022-09-22 02:45:12 +09:00
Jong Wook Kim
6e3be77e1a
initial commit
2022-09-22 01:09:43 +09:00
Jong Wook Kim
