[CI] Reorganize stage-b 1-GPU tests for 5090 compatibility (#16826)

test/registered/models/test_transformers_models.py

@@ -1,9 +1,4 @@
-from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
-
 # Transformers fallback model tests
-register_cuda_ci(est_time=245, suite="stage-b-test-small-1-gpu")
-register_cuda_ci(est_time=210, suite="stage-b-test-small-1-gpu-5090")
-register_amd_ci(est_time=320, suite="stage-b-test-small-1-gpu-amd")
 
 import dataclasses
 import multiprocessing as mp
@@ -14,6 +9,7 @@ from typing import List
 import torch
 
 from sglang.srt.utils import is_hip, kill_process_tree
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.runners import DEFAULT_PROMPTS, SRTRunner, check_close_model_outputs
 from sglang.test.test_utils import (
     DEFAULT_MODEL_NAME_FOR_TEST,
@@ -24,6 +20,9 @@ from sglang.test.test_utils import (
     popen_launch_server,
 )
 
+register_cuda_ci(est_time=245, suite="stage-b-test-small-1-gpu")
+register_amd_ci(est_time=320, suite="stage-b-test-small-1-gpu-amd")
+
 
 class TestTransformersFallbackEndpoint(CustomTestCase):
     @classmethod