refactor(multimodal/test): centralize model names and shared utilities in test_utils (#19354)

Co-authored-by: Ratish P <114130421+Ratish1@users.noreply.github.com>
2026-03-05 20:09:42 -08:00
parent 51e5dc845a
commit be9a9e4819
4 changed files with 221 additions and 43 deletions
@@ -27,8 +27,24 @@ from typing import Sequence

 from sglang.multimodal_gen.runtime.platforms import current_platform
 from sglang.multimodal_gen.runtime.utils.perf_logger import RequestPerfRecord
-
-DEFAULT_SMALL_MODEL = "Tongyi-MAI/Z-Image-Turbo"
+from sglang.multimodal_gen.test.test_utils import (
+    DEFAULT_FLUX_1_DEV_MODEL_NAME_FOR_TEST,
+    DEFAULT_FLUX_2_DEV_MODEL_NAME_FOR_TEST,
+    DEFAULT_FLUX_2_KLEIN_4B_MODEL_NAME_FOR_TEST,
+    DEFAULT_QWEN_IMAGE_EDIT_2509_MODEL_NAME_FOR_TEST,
+    DEFAULT_QWEN_IMAGE_EDIT_2511_MODEL_NAME_FOR_TEST,
+    DEFAULT_QWEN_IMAGE_EDIT_MODEL_NAME_FOR_TEST,
+    DEFAULT_QWEN_IMAGE_LAYERED_MODEL_NAME_FOR_TEST,
+    DEFAULT_QWEN_IMAGE_MODEL_NAME_FOR_TEST,
+    DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_1_I2V_14B_480P_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_1_I2V_14B_720P_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_1_T2V_14B_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_2_I2V_A14B_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_2_T2V_A14B_MODEL_NAME_FOR_TEST,
+    DEFAULT_WAN_2_2_TI2V_5B_MODEL_NAME_FOR_TEST,
+)


@dataclass
@@ -301,8 +317,6 @@ class PerformanceSummary:
        )


-SMALL_T2I_MODEL = "Tongyi-MAI/Z-Image-Turbo"
-
 T2I_sampling_params = DiffusionSamplingParams(
    prompt="Doraemon is eating dorayaki",
    output_size="1024x1024",
@@ -362,7 +376,7 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "qwen_image_t2i",
        DiffusionServerArgs(
-            model_path="Qwen/Qwen-Image",
+            model_path=DEFAULT_QWEN_IMAGE_MODEL_NAME_FOR_TEST,
            modality="image",
        ),
        T2I_sampling_params,
@@ -370,7 +384,7 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "qwen_image_t2i_cache_dit_enabled",
        DiffusionServerArgs(
-            model_path="Qwen/Qwen-Image",
+            model_path=DEFAULT_QWEN_IMAGE_MODEL_NAME_FOR_TEST,
            modality="image",
            enable_cache_dit=True,
        ),
@@ -379,7 +393,7 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "flux_image_t2i",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.1-dev", modality="image"
+            model_path=DEFAULT_FLUX_1_DEV_MODEL_NAME_FOR_TEST, modality="image"
        ),
        T2I_sampling_params,
    ),
@@ -396,14 +410,14 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "flux_2_image_t2i",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.2-dev", modality="image"
+            model_path=DEFAULT_FLUX_2_DEV_MODEL_NAME_FOR_TEST, modality="image"
        ),
        T2I_sampling_params,
    ),
    DiffusionTestCase(
        "flux_2_klein_image_t2i",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.2-klein-4B",
+            model_path=DEFAULT_FLUX_2_KLEIN_4B_MODEL_NAME_FOR_TEST,
            modality="image",
        ),
        T2I_sampling_params,
@@ -414,7 +428,7 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "layerwise_offload",
        DiffusionServerArgs(
-            model_path=SMALL_T2I_MODEL,
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
            modality="image",
            dit_layerwise_offload=True,
            dit_offload_prefetch_size=2,
@@ -423,13 +437,15 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    ),
    DiffusionTestCase(
        "zimage_image_t2i",
-        DiffusionServerArgs(model_path="Tongyi-MAI/Z-Image-Turbo", modality="image"),
+        DiffusionServerArgs(
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST, modality="image"
+        ),
        T2I_sampling_params,
    ),
    DiffusionTestCase(
        "zimage_image_t2i_fp8",
        DiffusionServerArgs(
-            model_path="Tongyi-MAI/Z-Image-Turbo",
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
            modality="image",
            extras=["--transformer-path MickJ/Z-Image-Turbo-fp8"],
        ),
@@ -439,7 +455,7 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "zimage_image_t2i_multi_lora",
        DiffusionServerArgs(
-            model_path="Tongyi-MAI/Z-Image-Turbo",
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
            modality="image",
            lora_path="reverentelusarca/elusarca-anime-style-lora-z-image-turbo",
            second_lora_path="tarn59/pixel_art_style_lora_z_image_turbo",
@@ -449,22 +465,33 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
    # === Text and Image to Image (TI2I) ===
    DiffusionTestCase(
        "qwen_image_edit_ti2i",
-        DiffusionServerArgs(model_path="Qwen/Qwen-Image-Edit", modality="image"),
+        DiffusionServerArgs(
+            model_path=DEFAULT_QWEN_IMAGE_EDIT_MODEL_NAME_FOR_TEST, modality="image"
+        ),
        TI2I_sampling_params,
    ),
    DiffusionTestCase(
        "qwen_image_edit_2509_ti2i",
-        DiffusionServerArgs(model_path="Qwen/Qwen-Image-Edit-2509", modality="image"),
+        DiffusionServerArgs(
+            model_path=DEFAULT_QWEN_IMAGE_EDIT_2509_MODEL_NAME_FOR_TEST,
+            modality="image",
+        ),
        MULTI_IMAGE_TI2I_sampling_params,
    ),
    DiffusionTestCase(
        "qwen_image_edit_2511_ti2i",
-        DiffusionServerArgs(model_path="Qwen/Qwen-Image-Edit-2511", modality="image"),
+        DiffusionServerArgs(
+            model_path=DEFAULT_QWEN_IMAGE_EDIT_2511_MODEL_NAME_FOR_TEST,
+            modality="image",
+        ),
        TI2I_sampling_params,
    ),
    DiffusionTestCase(
        "qwen_image_layered_i2i",
-        DiffusionServerArgs(model_path="Qwen/Qwen-Image-Layered", modality="image"),
+        DiffusionServerArgs(
+            model_path=DEFAULT_QWEN_IMAGE_LAYERED_MODEL_NAME_FOR_TEST,
+            modality="image",
+        ),
        MULTI_FRAME_I2I_sampling_params,
    ),
 ]
@@ -479,7 +506,7 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "wan2_1_t2v_1.3b",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
        ),
@@ -490,7 +517,7 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "wan2_1_t2v_1.3b_text_encoder_cpu_offload",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            text_encoder_cpu_offload=True,
@@ -503,7 +530,7 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "wan2_1_t2v_1.3b_teacache_enabled",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
        ),
@@ -533,7 +560,7 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "wan2_1_t2v_1_3b_lora_1gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=1,
@@ -558,14 +585,14 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "flux_2_ti2i",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.2-dev", modality="image"
+            model_path=DEFAULT_FLUX_2_DEV_MODEL_NAME_FOR_TEST, modality="image"
        ),
        TI2I_sampling_params,
    ),
    DiffusionTestCase(
        "flux_2_t2i_customized_vae_path",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.2-dev",
+            model_path=DEFAULT_FLUX_2_DEV_MODEL_NAME_FOR_TEST,
            modality="image",
            extras=["--vae-path=fal/FLUX.2-Tiny-AutoEncoder"],
        ),
@@ -587,7 +614,7 @@ ONE_GPU_CASES_B: list[DiffusionTestCase] = [
    DiffusionTestCase(
        "wan2_2_ti2v_5b",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.2-TI2V-5B-Diffusers",
+            model_path=DEFAULT_WAN_2_2_TI2V_5B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
        ),
@@ -636,7 +663,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "wan2_2_i2v_a14b_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.2-I2V-A14B-Diffusers",
+            model_path=DEFAULT_WAN_2_2_I2V_A14B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
        ),
@@ -645,7 +672,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "wan2_2_t2v_a14b_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.2-T2V-A14B-Diffusers",
+            model_path=DEFAULT_WAN_2_2_T2V_A14B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -658,7 +685,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "wan2_2_t2v_a14b_lora_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.2-T2V-A14B-Diffusers",
+            model_path=DEFAULT_WAN_2_2_T2V_A14B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -671,7 +698,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "wan2_1_t2v_14b_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-14B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_14B_MODEL_NAME_FOR_TEST,
            modality="video",
            num_gpus=2,
            custom_validator="video",
@@ -684,7 +711,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "wan2_1_t2v_1.3b_cfg_parallel",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
+            model_path=DEFAULT_WAN_2_1_T2V_1_3B_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -697,7 +724,7 @@ TWO_GPU_CASES_A = [
    DiffusionTestCase(
        "fsdp-inference",
        DiffusionServerArgs(
-            model_path=DEFAULT_SMALL_MODEL,
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
            modality="image",
            num_gpus=2,
            extras=["--use-fsdp-inference"],
@@ -710,7 +737,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "wan2_1_i2v_14b_480P_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
+            model_path=DEFAULT_WAN_2_1_I2V_14B_480P_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -721,7 +748,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "wan2_1_i2v_14b_lora_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-I2V-14B-720P-Diffusers",
+            model_path=DEFAULT_WAN_2_1_I2V_14B_720P_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -732,7 +759,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "wan2_1_i2v_14b_720P_2gpu",
        DiffusionServerArgs(
-            model_path="Wan-AI/Wan2.1-I2V-14B-720P-Diffusers",
+            model_path=DEFAULT_WAN_2_1_I2V_14B_720P_MODEL_NAME_FOR_TEST,
            modality="video",
            custom_validator="video",
            num_gpus=2,
@@ -742,7 +769,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "qwen_image_t2i_2_gpus",
        DiffusionServerArgs(
-            model_path="Qwen/Qwen-Image",
+            model_path=DEFAULT_QWEN_IMAGE_MODEL_NAME_FOR_TEST,
            modality="image",
            num_gpus=2,
            # test ring attn
@@ -754,7 +781,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "zimage_image_t2i_2_gpus",
        DiffusionServerArgs(
-            model_path="Tongyi-MAI/Z-Image-Turbo",
+            model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
            modality="image",
            num_gpus=2,
            ulysses_degree=2,
@@ -764,7 +791,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "flux_image_t2i_2_gpus",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.1-dev",
+            model_path=DEFAULT_FLUX_1_DEV_MODEL_NAME_FOR_TEST,
            modality="image",
            num_gpus=2,
        ),
@@ -773,7 +800,7 @@ TWO_GPU_CASES_B = [
    DiffusionTestCase(
        "flux_2_image_t2i_2_gpus",
        DiffusionServerArgs(
-            model_path="black-forest-labs/FLUX.2-dev",
+            model_path=DEFAULT_FLUX_2_DEV_MODEL_NAME_FOR_TEST,
            modality="image",
            num_gpus=2,
            tp_size=2,