Fix next sentence output (#8466)

2020-11-11 15:41:39 +01:00
parent da842e4e72
commit 069b63844c
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/modeling_tf_outputs.py
+++ b/src/transformers/modeling_tf_outputs.py
@@ -325,7 +325,7 @@ class TFNextSentencePredictorOutput(ModelOutput):
            heads.
    """

-    loss: tf.Tensor = None
+    loss: Optional[tf.Tensor] = None
    logits: tf.Tensor = None
    hidden_states: Optional[Tuple[tf.Tensor]] = None
    attentions: Optional[Tuple[tf.Tensor]] = None