Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "cf3cf304ca234ce70ffa223d7868e63133acd7ec"
Commit 3e1cd824 authored by Rémi Louf's avatar Rémi Louf
Browse files

fix stupid (re)naming issue

parent 81ee29ee
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
transformers/modeling_bert.py
View file @ 3e1cd824
...@@ -333,12 +333,12 @@ class BertLayer(nn.Module): ...@@ -333,12 +333,12 @@ class BertLayer(nn.Module):
if encoder_hidden_state: if encoder_hidden_state:
try: try:
crossattention_outputs = self.crossattention(attention_output, attention_mask, head_mask, encoder_hidden_state) attention_outputs = self.crossattention(attention_output, attention_mask, head_mask, encoder_hidden_state)
except AttributeError as ae: except AttributeError as ae:
raise ae("you need to set `is_encoder` to True in the configuration to instantiate an encoder layer") raise ae("you need to set `is_encoder` to True in the configuration to instantiate an encoder layer")
crossattention_output = crossattention_outputs[0] attention_output = attention_outputs[0]
intermediate_output = self.intermediate(crossattention_output) intermediate_output = self.intermediate(attention_output)
layer_output = self.output(intermediate_output, attention_output) layer_output = self.output(intermediate_output, attention_output)
outputs = (layer_output,) + attention_outputs[1:] # add attentions if we output them outputs = (layer_output,) + attention_outputs[1:] # add attentions if we output them
return outputs return outputs
......
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