File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed
paddlenlp/transformers/semantic_search Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -170,12 +170,18 @@ def forward(self,
170
170
query_attention_mask )
171
171
172
172
pos_title_cls_embedding = self .get_pooled_embedding (
173
- pos_title_input_ids , pos_title_token_type_ids ,
174
- pos_title_position_ids , pos_title_attention_mask )
173
+ pos_title_input_ids ,
174
+ pos_title_token_type_ids ,
175
+ pos_title_position_ids ,
176
+ pos_title_attention_mask ,
177
+ is_query = False )
175
178
176
179
neg_title_cls_embedding = self .get_pooled_embedding (
177
- neg_title_input_ids , neg_title_token_type_ids ,
178
- neg_title_position_ids , neg_title_attention_mask )
180
+ neg_title_input_ids ,
181
+ neg_title_token_type_ids ,
182
+ neg_title_position_ids ,
183
+ neg_title_attention_mask ,
184
+ is_query = False )
179
185
180
186
all_title_cls_embedding = paddle .concat (
181
187
x = [pos_title_cls_embedding , neg_title_cls_embedding ], axis = 0 )
You can’t perform that action at this time.
0 commit comments