transformers/docs/source/main_classes
Patrick von Platen a1bbcf3f6c
Refactoring the generate() function (#6949)
* first draft

* show design proposition for new generate method

* up

* make better readable

* make first version

* gpt2 tests pass

* make beam search for gpt2 work

* add first encoder-decoder code

* delete typo

* make t5 work

* save indermediate

* make bart work with beam search

* finish beam search bart / t5

* add default kwargs

* make more tests pass

* fix no bad words sampler

* some fixes and tests for all distribution processors

* fix test

* fix rag slow tests

* merge to master

* add nograd to generate

* make all slow tests pass

* speed up generate

* fix edge case bug

* small fix

* correct typo

* add type hints and docstrings

* fix typos in tests

* add beam search tests

* add tests for beam scorer

* fix test rag

* finish beam search tests

* move generation tests in seperate file

* fix generation tests

* more tests

* add aggressive generation tests

* fix tests

* add gpt2 sample test

* add more docstring

* add more docs

* finish doc strings

* apply some more of sylvains and sams comments

* fix some typos

* make fix copies

* apply lysandres and sylvains comments

* final corrections on examples

* small fix for reformer
2020-11-03 16:04:22 +01:00
..
callback.rst Add AzureML in integrations via dedicated callback (#8062) 2020-10-27 14:21:54 -04:00
configuration.rst Models doc (#7345) 2020-09-23 13:20:45 -04:00
logging.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
model.rst Refactoring the generate() function (#6949) 2020-11-03 16:04:22 +01:00
optimizer_schedules.rst Models doc (#7345) 2020-09-23 13:20:45 -04:00
output.rst Models doc (#7345) 2020-09-23 13:20:45 -04:00
pipelines.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
processors.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
tokenizer.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00
trainer.rst Doc styling (#8067) 2020-10-26 18:26:02 -04:00