From 91d249cd9b9a9dcc8b6592ed6ca7f8e88b004f26 Mon Sep 17 00:00:00 2001 From: Raayan Dhar <58057652+raayandhar@users.noreply.github.com> Date: Thu, 27 Nov 2025 20:58:17 -1000 Subject: [PATCH] fix: small changes to enable test_mrope.py (#14082) Signed-off-by: Raayan Dhar raayan.dhar@gmail.com --- test/srt/rotary_embedding/test_mrope.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/srt/rotary_embedding/test_mrope.py b/test/srt/rotary_embedding/test_mrope.py index 4fbfd0691..c9837f87f 100644 --- a/test/srt/rotary_embedding/test_mrope.py +++ b/test/srt/rotary_embedding/test_mrope.py @@ -7,6 +7,7 @@ from transformers import AutoConfig from transformers import __version__ as TRANSFORMERS_VERSION from sglang.srt.layers.rotary_embedding import get_rope +from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler from sglang.srt.utils import ( cpu_has_amx_support, is_cpu, @@ -85,6 +86,8 @@ def test_mrope( dtype: torch.dtype, num_tokens: int, ): + set_global_server_args_for_scheduler(ServerArgs(model_path="dummy")) + atol = model_info.atol rtol = model_info.rtol @@ -124,7 +127,7 @@ def test_mrope( num_tokens, num_heads, num_kv_heads, head_dim, max_position, dtype, device ) - query_native, key_native = mrope_helper_class.forward_native( + query_native, key_native = mrope_helper_class._forward_native( positions, query.clone(), key.clone(),