Sam Shleifer
|
66fa8ceaea
|
PegasusForConditionalGeneration (torch version) (#6340)
Co-authored-by: Jingqing Zhang <jingqing.zhang15@imperial.ac.uk>
|
2020-08-11 14:31:23 -04:00 |
|
Sylvain Gugger
|
d951c14ae4
|
Model output test (#6155)
* Use return_dict=True in all tests
* Formatting
|
2020-07-31 09:44:37 -04:00 |
|
Sam Shleifer
|
5abe50381a
|
Fix #6096: MBartTokenizer's mask token (#6098)
|
2020-07-28 18:27:58 -04:00 |
|
Sam Shleifer
|
838950ee44
|
[fix] mbart_en_ro_generate test now identical to fairseq (#5731)
|
2020-07-14 06:12:24 -04:00 |
|
Sam Shleifer
|
353b8f1e7a
|
Add mbart-large-cc25, support translation finetuning (#5129)
improve unittests for finetuning, especially w.r.t testing frozen parameters
fix freeze_embeds for T5
add streamlit setup.cfg
|
2020-07-07 13:23:01 -04:00 |
|