[diffusion] CI: create and refactor UT (#19619)
This commit is contained in:
@@ -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
|
||||
],
|
||||
|
||||
Reference in New Issue
Block a user