[diffusion] CI: exercise the default sp selection in CI (#33931)
This commit is contained in:
@@ -932,14 +932,12 @@ TWO_GPU_CASES = [
|
|||||||
"zimage_image_t2i_2_gpus",
|
"zimage_image_t2i_2_gpus",
|
||||||
DiffusionServerArgs(
|
DiffusionServerArgs(
|
||||||
model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
|
model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
|
||||||
ulysses_degree=2,
|
|
||||||
),
|
),
|
||||||
),
|
),
|
||||||
DiffusionTestCase(
|
DiffusionTestCase(
|
||||||
"zimage_image_t2i_2_gpus_non_square",
|
"zimage_image_t2i_2_gpus_non_square",
|
||||||
DiffusionServerArgs(
|
DiffusionServerArgs(
|
||||||
model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
|
model_path=DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
|
||||||
ulysses_degree=2,
|
|
||||||
),
|
),
|
||||||
DiffusionSamplingParams(
|
DiffusionSamplingParams(
|
||||||
prompt=T2I_sampling_params.prompt,
|
prompt=T2I_sampling_params.prompt,
|
||||||
|
|||||||
Reference in New Issue
Block a user