Fix next sentence output (huggingface#8466)

fabiocapsouza · Nov 15, 2020 · 3252521 · 3252521
1 parent 18bb1c7
commit 3252521
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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