zimage ¶

ZImagePipelineConfig(model_path: str = '', pipeline_config_path: str | None = None, embedded_cfg_scale: float = 0.0, flow_shift: float | None = 3.0, flow_shift_sr: float | None = None, disable_autocast: bool = False, scheduler_step_in_fp32: bool = True, is_causal: bool = False, dit_config: ZImageDiTConfig = ZImageDiTConfig(), dit_precision: str = 'bf16', upsampler_config: UpsamplerConfig = UpsamplerConfig(), upsampler_precision: str = 'fp32', vae_config: AutoencoderKLVAEConfig = AutoencoderKLVAEConfig(), vae_precision: str = 'fp32', vae_decode_precision: str | None = None, vae_tiling: bool = False, vae_sp: bool = False, image_encoder_config: EncoderConfig = EncoderConfig(), image_encoder_precision: str = 'fp32', text_encoder_configs: tuple[EncoderConfig, ...] = (lambda: (Qwen3TextConfig(chat_template_enable_thinking=True),))(), text_encoder_precisions: tuple[str, ...] = (lambda: ('bf16',))(), preprocess_text_funcs: tuple[Callable[[str], str], ...] = (lambda: (preprocess_text,))(), postprocess_text_funcs: tuple[Callable[[BaseEncoderOutput], Tensor], ...] = (lambda: (_zimage_text_postprocess,))(), dmd_denoising_steps: list[int] | None = None, ti2v_task: bool = False, lucy_edit_task: bool = False, boundary_ratio: float | None = None, scheduler_arch: str = 'FlowMatchEulerDiscreteScheduler', transformer_arch: str = 'ZImageTransformer2DModel', vae_arch: str = 'AutoencoderKL', text_encoder_archs: tuple[str, ...] = ('Qwen3Model',), tokenizer_archs: tuple[str, ...] = ('Qwen2Tokenizer',), scheduler_sigma_min: float = 0.0, scheduler_use_reference_discrete_timesteps: bool = True)