bi{ \ddlmZddlmZmZddlmZddlmZm Z m Z ddl m Z ddl mZdd lmZej"eZGd d eZGd d eZGddeZGddeZGddeZedefdefde fde fdefde fgZedefdefdefdefgZeedZy))logging)AutoPipelineBlocksSequentialPipelineBlocks)InsertableDict) WanInputStepWanPrepareLatentsStepWanSetTimestepsStep) WanDecodeStep)WanDenoiseStep)WanTextEncoderStepc.eZdZeeegZgdZedZ y)WanBeforeDenoiseStep)input set_timestepsprepare_latentsc y)NaBefore denoise step that prepare the inputs for the denoise step. This is a sequential pipeline blocks: - `WanInputStep` is used to adjust the batch size of the model inputs - `WanSetTimestepsStep` is used to set the timesteps - `WanPrepareLatentsStep` is used to prepare the latents selfs i/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/modular_pipelines/wan/modular_blocks.py descriptionz WanBeforeDenoiseStep.description(s  L N) __name__ __module__ __qualname__r r r block_classes block_namespropertyrrrrrr s,M @K   rrc.eZdZegZdgZdgZedZy)WanAutoBeforeDenoiseSteptext2vidNc y)NzBefore denoise step that prepare the inputs for the denoise step. This is an auto pipeline block that works for text2vid. - `WanBeforeDenoiseStep` (text2vid) is used. rrs rrz$WanAutoBeforeDenoiseStep.description;  @ r) rrrrrrblock_trigger_inputsr rrrrr"r"4s/M,K 6   rr"c4eZdZegZdgZdgZedefdZ y)WanAutoDenoiseStepdenoiseNreturnc y)NzDenoise step that iteratively denoise the latents. This is a auto pipeline block that works for text2vid tasks.. - `WanDenoiseStep` (denoise) for text2vid tasks.rrs rrzWanAutoDenoiseStep.descriptionLr%r) rrrr rrr&r strrrrrr(r(Es6M+K 6  S  rr(c.eZdZegZdgZdgZedZy)WanAutoDecodeStepz non-inpaintNcy)NzTDecode step that decode the denoised latents into videos outputs. - `WanDecodeStep`rrs rrzWanAutoDecodeStep.description[sfr) rrrr rrr&r rrrrr.r.Vs,"OM /K 6 ggrr.c0eZdZeeeegZgdZe dZ y) WanAutoBlocks) text_encoderbefore_denoiser)decoderc y)NzvAuto Modular pipeline for text-to-video using Wan. - for text-to-video generation, all you need to provide is `prompt`rrs rrzWanAutoBlocks.descriptionos  T rN) rrrrr"r(r.rrr rrrrr1r1as0  M K  rr1r2rrrr)decoder3) text2videoautoN)utilsrmodular_pipelinerrmodular_pipeline_utilsrr3r r r decodersr r)r encodersr get_loggerrloggerrr"r(r.r1TEXT2VIDEO_BLOCKS AUTO_BLOCKS ALL_BLOCKSrrrrCsK3 $#(   H % 3 (  1  "  +  "g*g , ,# +, , -. 12 N# =!   +, 34 &' $%  $  r