SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Model Templates for Seq2Seq (#9251)

Browse Source 
* 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
This commit is contained in:
Patrick von Platen
2020-12-22 23:41:20 +01:00
committed by GitHub
parent e6c1f1cad8
commit cbe63949d7
21 changed files with 3561 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_tf_bart.py
View File

@@ -94,7 +94,7 @@ class TFBartModelTester:
self.batch_size = 1
# 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()
past_key_values = past_key_values[1]