Skip to content

Commit 7fab504

Browse files
authored
fix bugs on windows (#1465)
1 parent 87fa30b commit 7fab504

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

paddlenlp/taskflow/text_similarity.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -126,10 +126,10 @@ def _preprocess(self, inputs):
126126
]
127127

128128
batchify_fn = lambda samples, fn=Tuple(
129-
Pad(axis=0, pad_val=self._tokenizer.pad_token_id), # text1_input_ids
130-
Pad(axis=0, pad_val=self._tokenizer.pad_token_type_id), # text1_token_type_ids
131-
Pad(axis=0, pad_val=self._tokenizer.pad_token_id), # text2_input_ids
132-
Pad(axis=0, pad_val=self._tokenizer.pad_token_type_id), # text2_token_type_ids
129+
Pad(axis=0, pad_val=self._tokenizer.pad_token_id, dtype='int64'), # text1_input_ids
130+
Pad(axis=0, pad_val=self._tokenizer.pad_token_type_id, dtype='int64'), # text1_token_type_ids
131+
Pad(axis=0, pad_val=self._tokenizer.pad_token_id, dtype='int64'), # text2_input_ids
132+
Pad(axis=0, pad_val=self._tokenizer.pad_token_type_id, dtype='int64'), # text2_token_type_ids
133133
): [data for data in fn(samples)]
134134

135135
outputs = {}

0 commit comments

Comments
 (0)