From d881f31488b7e0005517a00b74e5d20b01c0cde5 Mon Sep 17 00:00:00 2001 From: Mick Date: Sat, 6 Dec 2025 12:39:37 +0800 Subject: [PATCH] [diffusion] chore: temporarily upgrade diffusers to make Z-image compatible with Cache-DiT (#14530) --- python/pyproject.toml | 3 ++- .../sglang/multimodal_gen/configs/sample/sampling_params.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index 56a2226b9..5318ce643 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -85,7 +85,8 @@ checkpoint-engine = ["checkpoint-engine==0.1.2"] diffusion = [ "PyYAML==6.0.1", "cloudpickle", - "diffusers==0.35.2", + # this is temporary + "diffusers @ git+https://github.com/huggingface/diffusers.git@6290fdfda40610ce7b99920146853614ba529c6e", "imageio==2.36.0", "imageio-ffmpeg==0.5.1", "moviepy>=2.0.0", diff --git a/python/sglang/multimodal_gen/configs/sample/sampling_params.py b/python/sglang/multimodal_gen/configs/sample/sampling_params.py index b05fb62e9..0e7563377 100644 --- a/python/sglang/multimodal_gen/configs/sample/sampling_params.py +++ b/python/sglang/multimodal_gen/configs/sample/sampling_params.py @@ -137,8 +137,8 @@ class SamplingParams: return_frames: bool = False return_trajectory_latents: bool = False # returns all latents for each timestep return_trajectory_decoded: bool = False # returns decoded latents for each timestep - # if True, allow user params to override subclass-defined protected fields - no_override_protected_fields: bool = True + # if True, disallow user params to override subclass-defined protected fields + no_override_protected_fields: bool = False # whether to adjust num_frames for multi-GPU friendly splitting (default: True) adjust_frames: bool = True