From 1f58fcf0d7a74f4d5440959bc5b7ac7beda1f26f Mon Sep 17 00:00:00 2001 From: Tcc0403 <76503978+Tcc0403@users.noreply.github.com> Date: Fri, 16 Jan 2026 20:51:25 +0800 Subject: [PATCH] Remove deprecated `pos_id` argument Signed-off-by: Tcc0403 <76503978+Tcc0403@users.noreply.github.com> --- test/transformers/test_rope.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/transformers/test_rope.py b/test/transformers/test_rope.py index a7623a236..4df7da938 100644 --- a/test/transformers/test_rope.py +++ b/test/transformers/test_rope.py @@ -83,7 +83,7 @@ def test_correctness( cos, sin = rotary_emb(k1, pos_ids) # validate forward pass - hf_q, hf_k = apply_rotary_pos_emb(q1, k1, cos, sin, pos_ids) + hf_q, hf_k = apply_rotary_pos_emb(q1, k1, cos, sin) tt_q, tt_k = liger_rotary_pos_emb(q2, k2, cos, sin) assert torch.allclose(hf_q, tt_q, atol=atol, rtol=rtol) assert torch.allclose(hf_k, tt_k, atol=atol, rtol=rtol)