Skip to content

Commit edceb13

Browse files
authored
fix the bug in text2sql (#828)
1 parent df763fa commit edceb13

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/text_to_sql/RAT-SQL/text2sql/models/encoder_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def forward(self, inputs):
112112
"""
113113
seq_hidden, cls_hidden = self.base_encoder(inputs['src_ids'],
114114
inputs['sent_ids'])
115-
if self.pretrain_model_type != 'ERNIE':
115+
if self.pretrain_model_type != 'ERNIE' and self.pretrain_model_type != 'BERT':
116116
cls_hidden, seq_hidden = seq_hidden, cls_hidden
117117

118118
question_tokens_index = inputs["question_tokens_index"]

0 commit comments

Comments
 (0)