Skip to content

GitLab

"tasks/git@developer.sourcefind.cn:wuxk1/megatron-lm.git" did not exist on "1016e98a8351e7b1fd630c6930aa4fbb491df895"
Unverified Commit 069b6384 authored by Julien Plu's avatar Julien Plu Committed by GitHub
Browse files

Fix next sentence output (#8466)

parent da842e4e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/modeling_tf_outputs.py
View file @ 069b6384
...@@ -325,7 +325,7 @@ class TFNextSentencePredictorOutput(ModelOutput): ...@@ -325,7 +325,7 @@ class TFNextSentencePredictorOutput(ModelOutput):
heads. heads.
""" """
loss: tf.Tensor = None loss: Optional[tf.Tensor] = None
logits: tf.Tensor = None logits: tf.Tensor = None
hidden_states: Optional[Tuple[tf.Tensor]] = None hidden_states: Optional[Tuple[tf.Tensor]] = None
attentions: Optional[Tuple[tf.Tensor]] = None attentions: Optional[Tuple[tf.Tensor]] = None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment