From 069b63844c9255cb0a56c632de281a3688834190 Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Wed, 11 Nov 2020 15:41:39 +0100 Subject: [PATCH] Fix next sentence output (#8466) --- src/transformers/modeling_tf_outputs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_tf_outputs.py b/src/transformers/modeling_tf_outputs.py index 8d7d53814c..e2c6ceb2c0 100644 --- 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