We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent df763fa commit edceb13Copy full SHA for edceb13
examples/text_to_sql/RAT-SQL/text2sql/models/encoder_v2.py
@@ -112,7 +112,7 @@ def forward(self, inputs):
112
"""
113
seq_hidden, cls_hidden = self.base_encoder(inputs['src_ids'],
114
inputs['sent_ids'])
115
- if self.pretrain_model_type != 'ERNIE':
+ if self.pretrain_model_type != 'ERNIE' and self.pretrain_model_type != 'BERT':
116
cls_hidden, seq_hidden = seq_hidden, cls_hidden
117
118
question_tokens_index = inputs["question_tokens_index"]
0 commit comments