chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] fix: replace TextEncoderConfig with Qwen3TextConfig for Z-Image (#18560)

Browse Source 
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Vinh H. Pham
2026-02-12 13:31:23 +07:00
committed by GitHub
parent c9297661b9
commit feaa9e7e00
2 changed files with 12 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/sglang/multimodal_gen/configs/pipeline_configs/zimage.py
View File

@@ -7,10 +7,8 @@ import torch
from sglang.multimodal_gen.configs.models import DiTConfig, EncoderConfig, VAEConfig
from sglang.multimodal_gen.configs.models.dits.zimage import ZImageDitConfig
from sglang.multimodal_gen.configs.models.encoders import (
BaseEncoderOutput,
TextEncoderConfig,
)
from sglang.multimodal_gen.configs.models.encoders import BaseEncoderOutput
from sglang.multimodal_gen.configs.models.encoders.qwen3 import Qwen3TextConfig
from sglang.multimodal_gen.configs.models.vaes.flux import FluxVAEConfig
from sglang.multimodal_gen.configs.pipeline_configs.base import (
ImagePipelineConfig,
@@ -49,7 +47,7 @@ class ZImagePipelineConfig(ImagePipelineConfig):
dit_config: DiTConfig = field(default_factory=ZImageDitConfig)
vae_config: VAEConfig = field(default_factory=FluxVAEConfig)
text_encoder_configs: tuple[EncoderConfig, ...] = field(
default_factory=lambda: (TextEncoderConfig(),)
default_factory=lambda: (Qwen3TextConfig(),)
)
preprocess_text_funcs: tuple[Callable, ...] = field(

18
python/sglang/multimodal_gen/test/server/perf_baselines.json
View File

@@ -525,7 +525,7 @@
"stages_ms": {
"ConditioningStage": 0.01,
"DecodingStage": 33.93,
"TextEncodingStage": 104.21,
"TextEncodingStage": 409.74,
"InputValidationStage": 0.03,
"DenoisingStage": 808.69,
"LatentPreparationStage": 0.14,
@@ -542,14 +542,14 @@
"7": 99.3,
"8": 100.83
},
"expected_e2e_ms": 952.09,
"expected_e2e_ms": 1461.08,
"expected_avg_denoise_ms": 89.36,
"expected_median_denoise_ms": 99.67
},
"zimage_image_t2i_multi_lora": {
"stages_ms": {
"InputValidationStage": 0.04,
"TextEncodingStage": 103.81,
"TextEncodingStage": 413.69,
"ConditioningStage": 0.01,
"TimestepPreparationStage": 1.3,
"LatentPreparationStage": 0.11,
@@ -567,14 +567,14 @@
"7": 100.86,
"8": 103.87
},
"expected_e2e_ms": 955.14,
"expected_e2e_ms": 1464.31,
"expected_avg_denoise_ms": 89.96,
"expected_median_denoise_ms": 99.72
},
"zimage_image_t2i_2_gpus": {
"stages_ms": {
"InputValidationStage": 0.08,
"TextEncodingStage": 118.66,
"TextEncodingStage": 420.74,
"ConditioningStage": 0.01,
"TimestepPreparationStage": 1.5,
"LatentPreparationStage": 0.12,
@@ -599,7 +599,7 @@
"zimage_image_t2i_warmup": {
"stages_ms": {
"InputValidationStage": 0.02,
"TextEncodingStage": 100.65,
"TextEncodingStage": 456.16,
"ConditioningStage": 0.01,
"TimestepPreparationStage": 0.98,
"LatentPreparationStage": 0.06,
@@ -617,7 +617,7 @@
"7": 110.54,
"8": 115.24
},
"expected_e2e_ms": 1029.96,
"expected_e2e_ms": 1184.45,
"expected_avg_denoise_ms": 98.46,
"expected_median_denoise_ms": 109.65
},
@@ -2001,7 +2001,7 @@
"fsdp-inference": {
"stages_ms": {
"InputValidationStage": 0.04,
"TextEncodingStage": 128.3,
"TextEncodingStage": 411.12,
"ConditioningStage": 0.01,
"TimestepPreparationStage": 1.44,
"LatentPreparationStage": 0.1,
@@ -2019,7 +2019,7 @@
"7": 178.53,
"8": 178.08
},
"expected_e2e_ms": 1742.7,
"expected_e2e_ms": 2103.05,
"expected_avg_denoise_ms": 173.83,
"expected_median_denoise_ms": 178.08
}