Add output in a dictionary for TF `generate` method #12139
Add output args to greedy search
cfc32a57
Fix critical typo + make style quality
a977b75e
Handle generate_beam_search
82bbcdad
Add dict_specific tests and fix the placement of encoder outputs
6f0cadf2
Add specific outputs
4fbd6563
Update doc
f4b2b214
Fix typo
574f8327
Merge master into tf_generate_output_utils
5771ae33
Adjust handling encoder_outputs + Fix generating for T5
bf59a287
Fix generate for RAG
078fc0c4
Fix handling ouptut_attentions when target_mapping is not None
81cfb0e4
stancld
changed the title Add output in a dictionary for TF `generate` method [WIP] Add output in a dictionary for TF `generate` method 4 years ago
Rename kwargs to model_kwargs
d3d2911e
make style quality
ca045129
Merge master into tf_generate_output_utils
767ba0ee
Move imports in test_modeling_tf_common.py
84e357f3
stancld
changed the title [WIP] Add output in a dictionary for TF `generate` method Add output in a dictionary for TF `generate` method 4 years ago
Merge master into tf_generate_output_utils
4a266342
Rewrite nested if-statements
4419c708
sgugger
approved these changes
on 2021-06-17
Fix added tests
11d4fd25
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub