transformers/docs/source/internal
Patrick von Platen c3c39f7e84
[Flax] Add Beam Search (#12131)
* fix_torch_device_generate_test

* remove @

* push new logit processors

* add processors

* save first working version

* save intermediate

* finish

* make style

* make fix-copies

* finish

* Update tests/test_modeling_flax_bart.py

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Suraj Patil <surajp815@gmail.com>

Co-authored-by: Patrick von Platen <patrick@huggingface.co>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
2021-06-16 09:43:54 +01:00
..
file_utils.rst [PretrainedFeatureExtractor] + Wav2Vec2FeatureExtractor, Wav2Vec2Processor, Wav2Vec2Tokenizer (#10324) 2021-02-25 17:42:46 +03:00
generation_utils.rst [Flax] Add Beam Search (#12131) 2021-06-16 09:43:54 +01:00
modeling_utils.rst Fix T5 and BART for TF (#9063) 2020-12-14 18:47:00 +01:00
pipelines_utils.rst Instantiate model only once in pipeline (#10888) 2021-03-29 10:39:14 -04:00
tokenization_utils.rst [PretrainedFeatureExtractor] + Wav2Vec2FeatureExtractor, Wav2Vec2Processor, Wav2Vec2Tokenizer (#10324) 2021-02-25 17:42:46 +03:00
trainer_utils.rst [debug utils] activation/weights underflow/overflow detector (#11274) 2021-04-30 11:15:46 -07:00