Unverified Commit cbe63949 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Model Templates for Seq2Seq (#9251)

* adapt cookie cutter

* fix copy past statement

* delete copy statements for now

* remove unused import from template

* make doc rst

* correct config docstring

* correct training

* correct inputs processing tf enc dec

* make style

* adapt templates

* clean tabs

* correct tensor -> Tensor naming

* correct indent

* correct templates

* fix the test

* break lines to avoid > 119

* Apply suggestions from code review
parent e6c1f1ca
...@@ -94,7 +94,7 @@ class TFBartModelTester: ...@@ -94,7 +94,7 @@ class TFBartModelTester:
self.batch_size = 1 self.batch_size = 1
# first forward pass # first forward pass
outputs = model(input_ids, use_cache=True) outputs = model(input_ids, attention_mask=attention_mask, use_cache=True)
output, past_key_values = outputs.to_tuple() output, past_key_values = outputs.to_tuple()
past_key_values = past_key_values[1] past_key_values = past_key_values[1]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment