transformers/docs/source/internal
Patrick von Platen 2e12b907ae
TF generate refactor - Greedy Search (#15562)
* TF generate start refactor

* Add tf tests for sample generate

* re-organize

* boom boom

* Apply suggestions from code review

* re-add

* add all code

* make random greedy pass

* make encoder-decoder random work

* further improvements

* delete bogus file

* make gpt2 and t5 tests work

* finish logits tests

* correct logits processors

* correct past / encoder_outputs drama

* refactor some methods

* another fix

* refactor shape_list

* fix more shape list

* import shape
_list

* finish docs

* fix imports

* make style

* correct tf utils

* Fix TFRag as well

* Apply Lysandre's and Sylvais suggestions

* Update tests/test_generation_tf_logits_process.py

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

* Update src/transformers/tf_utils.py

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

* remove cpu according to gante

* correct logit processor

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
2022-02-15 17:54:43 +01:00
..
file_utils.mdx Convert rst files (#14888) 2021-12-22 16:14:35 -05:00
generation_utils.mdx TF generate refactor - Greedy Search (#15562) 2022-02-15 17:54:43 +01:00
modeling_utils.mdx Convert rst files (#14888) 2021-12-22 16:14:35 -05:00
pipelines_utils.mdx Convert rst files (#14888) 2021-12-22 16:14:35 -05:00
tokenization_utils.mdx Convert rst files (#14888) 2021-12-22 16:14:35 -05:00
trainer_utils.mdx Convert rst files (#14888) 2021-12-22 16:14:35 -05:00