[Model]Add new nextstep_1(Diffusion) model(only T2I) (#612)

Signed-off-by: Dong Wang <dongw2019@gmail.com>
Signed-off-by: sniper35 <dongw2019@gmail.com>
Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Dong WandHongsheng Liu authored and GitHub committed 2026-02-26 00:09:11 +08:00
1 parent 466a7f7a62
commit 1eebb374f1
17 files changed
+2795 -42

No files matched your search

@@ -456,11 +456,11 @@ def maybe_wrap_vae_decode_with_patch_parallelism(
vae = getattr(pipeline, "vae", None)
if vae is None or not hasattr(vae, "decode"):
return
try:
from diffusers.models.autoencoders import AutoencoderKL
except Exception:
return
if not isinstance(vae, AutoencoderKL):
# NOTE: Use capability checks (not strict diffusers type checks) because
# NextStep's custom VAE is compatible but not a diffusers AutoencoderKL.
# TODO(vae-pp): Replace duck-typing with an explicit VAE compatibility
# validator/protocol and add per-model integration tests.
if not hasattr(vae, "decoder"):
return
if getattr(vae, "_vllm_vae_patch_parallel_installed", False):