[diffusion] chore: temporarily upgrade diffusers to make Z-image compatible with Cache-DiT (#14530)

This commit is contained in:
Mick
2025-12-06 12:39:37 +08:00
committed by GitHub
parent 2ac5b98395
commit d881f31488
2 changed files with 4 additions and 3 deletions

View File

@@ -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",

View File

@@ -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