From 6f425758d103b5d91f63417fbae8a061f2e13229 Mon Sep 17 00:00:00 2001 From: Daniel Tu <57485952+danghoangnhan@users.noreply.github.com> Date: Tue, 30 Dec 2025 14:22:20 +0800 Subject: [PATCH] fix: replace numpy.where with paddle.where for tensor operations in ParseQHead (#17019) Co-authored-by: Lin Manhui --- ppocr/modeling/heads/rec_parseq_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/heads/rec_parseq_head.py b/ppocr/modeling/heads/rec_parseq_head.py index 572387e4a7..cf79c7ebea 100644 --- a/ppocr/modeling/heads/rec_parseq_head.py +++ b/ppocr/modeling/heads/rec_parseq_head.py @@ -365,7 +365,7 @@ class ParseQHead(nn.Layer): temp = paddle.triu( x=paddle.ones(shape=[num_steps, num_steps], dtype="bool"), diagonal=2 ) - posi = np.where(temp.cpu().numpy() == True) + posi = paddle.where(temp == True) query_mask[posi] = 0 bos = paddle.full(shape=(bs, 1), fill_value=self.bos_id).astype("int64") for i in range(self.refine_iters):