Fix next sentence output (#8466)

huggingface · Nov 11, 2020 · 069b638 · 069b638
1 parent da842e4
commit 069b638
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