diff --git a/test/registered/model_loading/test_load_weights_from_remote_instance.py b/test/registered/model_loading/test_load_weights_from_remote_instance.py index 6ab09594f..2fded2ab5 100644 --- a/test/registered/model_loading/test_load_weights_from_remote_instance.py +++ b/test/registered/model_loading/test_load_weights_from_remote_instance.py @@ -39,7 +39,7 @@ from sglang.utils import terminate_process mp.set_start_method("spawn", force=True) register_cuda_ci(est_time=145, stage="extra-a", runner_config="2-gpu-large") -register_amd_ci(est_time=72, suite="stage-b-test-2-gpu-large-amd") +register_amd_ci(est_time=72, stage="extra-a", runner_config="2-gpu-large-amd") def verify_params_close(params1, params2, error_msg):