TF - add unpack_inputs decorator for marian (#16226)

* add unpack_inputs decorator * small fix for attn_mask string Co-authored-by: Johannes Kolbe <johannes.kolbe@tech.better.team>

TF - add unpack_inputs decorator for marian (#16226)
* add unpack_inputs decorator * small fix for attn_mask string Co-authored-by: Johannes Kolbe <johannes.kolbe@tech.better.team>
1c1e377e · Johannes Kolbe · GitHub · 81643edd · 1c1e377e
Unverified Commit 1c1e377e authored Mar 17, 2022 by Johannes Kolbe Committed by GitHub Mar 17, 2022
Expand all Show whitespace changes
Inline Side-by-side

Showing with 121 additions and 220 deletions

src/transformers/models/marian/modeling_tf_marian.py src/transformers/models/marian/modeling_tf_marian.py +121 -220

No files found.
--- a/src/transformers/models/marian/modeling_tf_marian.py
+++ b/src/transformers/models/marian/modeling_tf_marian.py