[diffusion] CI: create and refactor UT (#19619)

2026-03-01 19:38:20 +08:00
parent 80a6b32703
commit a75840b373
4 changed files with 7 additions and 3 deletions
@@ -29,14 +29,18 @@ _UPDATE_WEIGHTS_MODEL_PAIR_IDS = (
 )
 SUITES = {
    # no GPU required; safe to run on any CPU-only runner
    "unit": [
        "../unit/test_sampling_params_validate.py",
        "../unit/test_storage.py",
        "../unit/test_lora_format_adapter.py",
        # add new unit tests here
    ],
    "1-gpu": [
        "test_server_a.py",
        "test_server_b.py",
        "test_lora_format_adapter.py",
        # cli test
        "../cli/test_generate_t2i_perf.py",
        # unit tests (no server needed)
        "../test_sampling_params_validate.py",
        "test_update_weights_from_disk.py",
        # add new 1-gpu test files here
    ],