diff --git a/test/registered/amd/disaggregation/test_disaggregation_basic.py b/test/registered/amd/disaggregation/test_disaggregation_basic.py index 8e00d2ba1..ece633fe8 100644 --- a/test/registered/amd/disaggregation/test_disaggregation_basic.py +++ b/test/registered/amd/disaggregation/test_disaggregation_basic.py @@ -13,7 +13,6 @@ from sglang.test.server_fixtures.disaggregation_fixture import ( PDDisaggregationServerBase, ) from sglang.test.test_utils import ( - DEFAULT_MODEL_NAME_FOR_TEST, DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH, popen_launch_pd_server, ) @@ -36,7 +35,7 @@ class TestDisaggregationAccuracy(PDDisaggregationServerBase): print("SGLANG_TEST_RDMA_DEVICE is not set! Running without RDMA.") cls.rdma_devices = [] - cls.model = DEFAULT_MODEL_NAME_FOR_TEST + cls.model = "Qwen/Qwen3-8B" # DEFAULT_MODEL_NAME_FOR_TEST # Non blocking start servers @@ -235,7 +234,7 @@ class TestDisaggregationMooncakeFailure(PDDisaggregationServerBase): # set DISAGGREGATION_TEST_FAILURE_PROB to simulate failure os.environ["DISAGGREGATION_TEST_FAILURE_PROB"] = "0.05" - cls.model = DEFAULT_MODEL_NAME_FOR_TEST + cls.model = "Qwen/Qwen3-8B" # Non blocking start servers cls.start_prefill() @@ -347,7 +346,7 @@ class TestDisaggregationSimulatedRetract(PDDisaggregationServerBase): cls.rdma_devices = [] os.environ["SGLANG_TEST_RETRACT"] = "true" - cls.model = DEFAULT_MODEL_NAME_FOR_TEST + cls.model = "Qwen/Qwen3-8B" # Non blocking start servers cls.start_prefill() diff --git a/test/registered/amd/disaggregation/test_disaggregation_pp.py b/test/registered/amd/disaggregation/test_disaggregation_pp.py index 7f0d9ca87..a817ae78b 100644 --- a/test/registered/amd/disaggregation/test_disaggregation_pp.py +++ b/test/registered/amd/disaggregation/test_disaggregation_pp.py @@ -9,7 +9,6 @@ from sglang.test.server_fixtures.disaggregation_fixture import ( PDDisaggregationServerBase, ) from sglang.test.test_utils import ( - DEFAULT_MODEL_NAME_FOR_TEST, DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH, popen_launch_pd_server, try_cached_model, @@ -33,7 +32,7 @@ class TestDisaggregationPrefillPPAccuracy(PDDisaggregationServerBase): print("SGLANG_TEST_RDMA_DEVICE is not set! Running without RDMA.") cls.rdma_devices = [] - cls.model = try_cached_model(DEFAULT_MODEL_NAME_FOR_TEST) + cls.model = try_cached_model("Qwen/Qwen3-8B") # Non blocking start servers cls.start_prefill() @@ -126,7 +125,7 @@ class TestDisaggregationPrefillPPDynamicChunkAccuracy(PDDisaggregationServerBase print("SGLANG_TEST_RDMA_DEVICE is not set! Running without RDMA.") cls.rdma_devices = [] - cls.model = try_cached_model(DEFAULT_MODEL_NAME_FOR_TEST) + cls.model = try_cached_model("Qwen/Qwen3-8B") # Non blocking start servers cls.start_prefill() @@ -220,7 +219,7 @@ class TestDisaggregationDecodePPAccuracy(PDDisaggregationServerBase): print("SGLANG_TEST_RDMA_DEVICE is not set! Running without RDMA.") cls.rdma_devices = [] - cls.model = try_cached_model(DEFAULT_MODEL_NAME_FOR_TEST) + cls.model = try_cached_model("Qwen/Qwen3-8B") # Non blocking start servers cls.start_prefill()