[diffusion] quant: support quant for all dits (#19156)

Co-authored-by: zyzshishui <zyzshishui@gmail.com>
This commit is contained in:
Mick
2026-02-24 14:20:54 +08:00
committed by GitHub
co-authored by zyzshishui
parent 4f25a48d7a
commit 2e053d6eb6
19 changed files with 523 additions and 133 deletions
@@ -295,6 +295,8 @@ class PerformanceSummary:
)
SMALL_T2I_MODEL = "Tongyi-MAI/Z-Image-Turbo"
T2I_sampling_params = DiffusionSamplingParams(
prompt="Doraemon is eating dorayaki",
output_size="1024x1024",
@@ -397,9 +399,9 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
# TODO: currently, we don't support sending more than one request in test, and setting `num_outputs_per_prompt` to 2 doesn't guarantee the denoising be executed twice,
# so we do one warmup and send one request instead
DiffusionTestCase(
"flux_2_image_t2i_layerwise_offload",
"layerwise_offload",
DiffusionServerArgs(
model_path="black-forest-labs/FLUX.2-dev",
model_path=SMALL_T2I_MODEL,
modality="image",
dit_layerwise_offload=True,
dit_offload_prefetch_size=2,
@@ -411,6 +413,15 @@ ONE_GPU_CASES_A: list[DiffusionTestCase] = [
DiffusionServerArgs(model_path="Tongyi-MAI/Z-Image-Turbo", modality="image"),
T2I_sampling_params,
),
DiffusionTestCase(
"zimage_image_t2i_fp8",
DiffusionServerArgs(
model_path="Tongyi-MAI/Z-Image-Turbo",
modality="image",
extras=["--transformer-path MickJ/Z-Image-Turbo-fp8"],
),
T2I_sampling_params,
),
# Multi-LoRA test case for Z-Image-Turbo
DiffusionTestCase(
"zimage_image_t2i_multi_lora",