Patrick von Platen
996a315e76
Flax Generate (#11777)
* fix_torch_device_generate_test
* remove @
* add
* indexing
* correct a couple of tests
* fix tests
* add logits processor
* finish top_k, top_p, temp
* add docs
* correct flax prng key default
* improve generate
* add generation docs
* add docs
* make style
* revert model outputs change
* make style
* correct typo
* fix tests
* fix slow test
* add raise
* finish generation
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
2021-05-27 00:18:17 +01:00
..
2021-04-05 12:27:23 -04:00
2020-12-07 18:36:34 -05:00
2021-04-13 12:14:25 -04:00
2021-03-26 11:23:56 -04:00
2021-02-09 10:27:49 -05:00
2021-05-27 00:18:17 +01:00
2020-12-22 11:33:44 -05:00
2021-04-26 22:18:37 -04:00
2021-05-13 10:34:14 -04:00
2021-04-21 11:11:20 -04:00
2021-04-05 10:51:16 -04:00
2021-05-04 14:17:11 -07:00