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