anodyine/vllm-omni
1
0 Fork 0
Code Issues 4 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Use text encoder FP8 for Z-Image quality gate

Browse Source 
Signed-off-by: WeiQing Chen <david6666666@users.noreply.github.com>
This commit is contained in:
WeiQing Chen committed 2026-05-28 08:52:37 +00:00
1 parent eb0c4b1de4
commit 8593781a07
7 files changed
+42 -77
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tests/diffusion/quantization/test_quantization_quality.py
+27 -17
View File
@@ -119,23 +119,33 @@ QUALITY_CONFIGS = [
id="fp8_z_image",
model="Tongyi-MAI/Z-Image-Turbo",
quantization={
"transformer": {
"method": "fp8",
"ignored_layers": [
"img_mlp",
*[
f"layers.{layer_id}.{suffix}"
for layer_id in range(15, 30)
for suffix in (
"attention.to_qkv",
"attention.to_out.0",
"feed_forward.w13",
"feed_forward.w2",
)
],
"method": "fp8",
"ignored_layers": [
"img_mlp",
*[
f"layers.{layer_id}.{suffix}"
for layer_id in range(15, 30)
for suffix in (
"attention.to_qkv",
"attention.to_out.0",
"feed_forward.w13",
"feed_forward.w2",
)
],
},
"default": None,
*[
f"model.layers.{layer_id}.{suffix}"
for layer_id in range(28, 36)
for suffix in (
"self_attn.q_proj",
"self_attn.k_proj",
"self_attn.v_proj",
"self_attn.o_proj",
"mlp.gate_proj",
"mlp.up_proj",
"mlp.down_proj",
)
],
],
},
task="t2i",
prompt=(
@@ -159,7 +169,7 @@ QUALITY_CONFIGS = [
"starlight waterfalls. Ethereal, melancholic, and transcendent mood—like a moment of quiet revelation "
"at the edge of existence."
),
max_lpips=0.10,
max_lpips=0.15,
num_inference_steps=20,
),
QualityTestConfig(