[CI] Fix stale CP-v1 mock patches in NextN mm-embed test (#38280)
This commit is contained in:
@@ -56,10 +56,6 @@ def _make_model_nextn(vocab_size: int, hidden_size: int):
|
|||||||
model.rot_weight = None
|
model.rot_weight = None
|
||||||
model.alt_stream = None
|
model.alt_stream = None
|
||||||
model.quant_config = None
|
model.quant_config = None
|
||||||
model.cp_rank = None
|
|
||||||
model.cp_size = None
|
|
||||||
model.dsa_enable_prefill_cp = False
|
|
||||||
model.mla_enable_prefill_cp = False
|
|
||||||
model.mtp_block = MagicMock(side_effect=lambda **kw: (kw["hidden_states"], None))
|
model.mtp_block = MagicMock(side_effect=lambda **kw: (kw["hidden_states"], None))
|
||||||
return model
|
return model
|
||||||
|
|
||||||
@@ -96,9 +92,6 @@ class TestDeepseekNextNMmEmbed(CustomTestCase):
|
|||||||
object.__setattr__(model, "embed_tokens", mock_embed)
|
object.__setattr__(model, "embed_tokens", mock_embed)
|
||||||
|
|
||||||
with (
|
with (
|
||||||
patch(
|
|
||||||
"sglang.srt.models.deepseek_nextn.is_cp_v2_active", return_value=False
|
|
||||||
),
|
|
||||||
patch(
|
patch(
|
||||||
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
|
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
|
||||||
return_value=False,
|
return_value=False,
|
||||||
@@ -164,9 +157,6 @@ class TestDeepseekNextNMmEmbed(CustomTestCase):
|
|||||||
embed_calls = mock_embed.call_args_list
|
embed_calls = mock_embed.call_args_list
|
||||||
|
|
||||||
with (
|
with (
|
||||||
patch(
|
|
||||||
"sglang.srt.models.deepseek_nextn.is_cp_v2_active", return_value=False
|
|
||||||
),
|
|
||||||
patch(
|
patch(
|
||||||
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
|
"sglang.srt.models.deepseek_nextn.dsa_use_prefill_cp",
|
||||||
return_value=False,
|
return_value=False,
|
||||||
|
|||||||
Reference in New Issue
Block a user