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

test/registered/models/test_vlm_models.py

@@ -1,9 +1,3 @@
-from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
-
-# VLM (Vision Language Model) tests
-register_cuda_ci(est_time=228, suite="stage-b-test-small-1-gpu")
-register_amd_ci(est_time=420, suite="stage-b-test-small-1-gpu-amd")
-
 import argparse
 import random
 import sys
@@ -11,12 +5,19 @@ import unittest
 from types import SimpleNamespace
 
 from sglang.srt.utils import is_hip
+from sglang.test.ci.ci_register import register_amd_ci, register_cuda_ci
 from sglang.test.kits.mmmu_vlm_kit import (
     DEFAULT_MEM_FRACTION_STATIC,
     MMMUMultiModelTestBase,
 )
 from sglang.test.test_utils import is_in_ci
 
+# VLM (Vision Language Model) tests
+
+
+register_cuda_ci(est_time=228, suite="stage-b-test-large-1-gpu")
+register_amd_ci(est_time=420, suite="stage-b-test-small-1-gpu-amd")
+
 _is_hip = is_hip()
 # VLM models for testing
 if _is_hip: