c-my/transformers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix TF Longformer (#8460)

Browse Source
This commit is contained in:
Julien Plu
2020-11-11 12:54:15 +01:00
committed by GitHub
parent 8dda9167de
commit 23290836c3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/transformers/modeling_tf_longformer.py
+2 -2
View File
@@ -93,7 +93,7 @@ class TFLongformerBaseModelOutput(ModelOutput):
in the sequence.
"""
last_hidden_state: tf.Tensor
last_hidden_state: tf.Tensor = None
hidden_states: Optional[Tuple[tf.Tensor]] = None
attentions: Optional[Tuple[tf.Tensor]] = None
global_attentions: Optional[Tuple[tf.Tensor]] = None
@@ -140,7 +140,7 @@ class TFLongformerBaseModelOutputWithPooling(ModelOutput):
in the sequence.
"""
last_hidden_state: tf.Tensor
last_hidden_state: tf.Tensor = None
pooler_output: tf.Tensor = None
hidden_states: Optional[Tuple[tf.Tensor]] = None
attentions: Optional[Tuple[tf.Tensor]] = None