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

This commit is contained in:
Mick
2026-03-01 19:38:20 +08:00
committed by GitHub
parent 80a6b32703
commit a75840b373
4 changed files with 7 additions and 3 deletions

View File

@@ -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
],