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 5401f01 commit 920e1e8Copy full SHA for 920e1e8
paddlenlp/taskflow/text_similarity.py
@@ -220,9 +220,11 @@ def _preprocess(self, inputs):
220
]
221
if ("rocketqa" in self.model_name):
222
batchify_fn = lambda samples, fn=Tuple(
223
- Pad(axis=0, pad_val=self._tokenizer.pad_token_id), # input ids
224
- Pad(axis=0, pad_val=self._tokenizer.pad_token_type_id
225
- ), # token type ids
+ Pad(axis=0, pad_val=self._tokenizer.pad_token_id, dtype='int64'
+ ), # input ids
+ Pad(axis=0,
226
+ pad_val=self._tokenizer.pad_token_type_id,
227
+ dtype='int64'), # token type ids
228
): [data for data in fn(samples)]
229
else:
230
0 commit comments