Fix next sentence output (#8466)

069b6384 · Julien Plu · GitHub · da842e4e · 069b6384
Unverified Commit 069b6384 authored Nov 11, 2020 by Julien Plu Committed by GitHub Nov 11, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_tf_outputs.py src/transformers/modeling_tf_outputs.py +1 -1

No files found.
--- 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