fix: replace numpy.where with paddle.where for tensor operations in ParseQHead (#17019)

Co-authored-by: Lin Manhui <mhlin425@whu.edu.cn>
2026-09-17 17:44:52 +08:00 · 2025-12-30 14:22:20 +08:00
parent ea3fbcc599
commit 6f425758d1
1 changed files with 1 additions and 1 deletions
@@ -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):