transformers
Add output in a dictionary for TF `generate` method
#12139
Merged

Add output in a dictionary for TF `generate` method #12139

stancld
stancld Add output args to greedy search
cfc32a57
stancld Fix critical typo + make style quality
a977b75e
stancld Handle generate_beam_search
82bbcdad
stancld Add dict_specific tests and fix the placement of encoder outputs
6f0cadf2
stancld Add specific outputs
4fbd6563
stancld Update doc
f4b2b214
stancld Fix typo
574f8327
stancld Merge master into tf_generate_output_utils
5771ae33
stancld Adjust handling encoder_outputs + Fix generating for T5
bf59a287
stancld Fix generate for RAG
078fc0c4
stancld Fix handling ouptut_attentions when target_mapping is not None
81cfb0e4
stancld 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
stancld Rename kwargs to model_kwargs
d3d2911e
stancld make style quality
ca045129
stancld Merge master into tf_generate_output_utils
767ba0ee
stancld Move imports in test_modeling_tf_common.py
84e357f3
stancld 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
patrickvonplaten
patrickvonplaten commented on 2021-06-17
stancld Merge master into tf_generate_output_utils
4a266342
stancld Rewrite nested if-statements
4419c708
patrickvonplaten
patrickvonplaten commented on 2021-06-17
patrickvonplaten
patrickvonplaten approved these changes on 2021-06-17
sgugger
sgugger approved these changes on 2021-06-17
stancld Fix added tests
11d4fd25
patrickvonplaten
patrickvonplaten patrickvonplaten merged 26a2e365 into master 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone