bi& ddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZdd lmZmZej(eZGd d eZGd d eZGddeZGddeZGddeZGddeZGddeZedefde fde fde fdefdefgZedefdefde fde fdefdefdefgZedefdefdefdefdefgZ eee d Z!y!)")logging)AutoPipelineBlocksSequentialPipelineBlocks)InsertableDict)FluxImg2ImgPrepareLatentsStepFluxImg2ImgSetTimestepsStep FluxInputStepFluxPrepareLatentsStepFluxSetTimestepsStep)FluxDecodeStep)FluxDenoiseStep)FluxTextEncoderStepFluxVaeEncoderStepc.eZdZegZdgZdgZedZy)FluxAutoVaeEncoderStepimg2imgimagec y)NaVae encoder step that encode the image inputs into their latent representations. This is an auto pipeline block that works for img2img tasks. - `FluxVaeEncoderStep` (img2img) is used when only `image` is provided. - if `image` is provided, step will be skipped.selfs j/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/modular_pipelines/flux/modular_blocks.py descriptionz"FluxAutoVaeEncoderStep.description's  A N) __name__ __module__ __qualname__r block_classes block_namesblock_trigger_inputspropertyrrrrrr"s+'(M+K#9   rrc.eZdZeeegZgdZedZ y)FluxBeforeDenoiseStep)inputprepare_latents set_timestepsc y)Na"Before denoise step that prepare the inputs for the denoise step. This is a sequential pipeline blocks: - `FluxInputStep` is used to adjust the batch size of the model inputs - `FluxPrepareLatentsStep` is used to prepare the latents - `FluxSetTimestepsStep` is used to set the timesteps rrs rrz!FluxBeforeDenoiseStep.description:s  I rN) rrrr r r r r!r#rrrrr%r%2s,M @K   rr%c.eZdZeeegZgdZedZ y)FluxImg2ImgBeforeDenoiseStep)r&r(r'c y)NaABefore denoise step that prepare the inputs for the denoise step for img2img task. This is a sequential pipeline blocks: - `FluxInputStep` is used to adjust the batch size of the model inputs - `FluxImg2ImgSetTimestepsStep` is used to set the timesteps - `FluxImg2ImgPrepareLatentsStep` is used to prepare the latents rrs rrz(FluxImg2ImgBeforeDenoiseStep.descriptionJs  T rN) rrrr r r r r!r#rrrrr+r+Fs'"$?A^_M?K   rr+c4eZdZeegZddgZddgZedZ y)FluxAutoBeforeDenoiseStep text2imagerN image_latentsc y)NaBefore denoise step that prepare the inputs for the denoise step. This is an auto pipeline block that works for text2image. - `FluxBeforeDenoiseStep` (text2image) is used. - `FluxImg2ImgBeforeDenoiseStep` (img2img) is used when only `image_latents` is provided. rrs rrz%FluxAutoBeforeDenoiseStep.description[s  m r) rrrr%r+r r!r"r#rrrrr.r.Vs4*,HIM+K /2   rr.c4eZdZegZdgZdgZedefdZ y)FluxAutoDenoiseStepdenoiseNreturnc y)NzDenoise step that iteratively denoise the latents. This is a auto pipeline block that works for text2image and img2img tasks. - `FluxDenoiseStep` (denoise) for text2image and img2img tasks.rrs rrzFluxAutoDenoiseStep.descriptionks  O r) rrrrr r!r"r#strrrrrr3r3fs2$%M+K 6  S  rr3c.eZdZegZdgZdgZedZy)FluxAutoDecodeStepz non-inpaintNcy)NzTDecode step that decode the denoised latents into image outputs. - `FluxDecodeStep`rrs rrzFluxAutoDecodeStep.descriptionzsfr) rrrrr r!r"r#rrrrr9r9us-#$M /K 6 ggrr9c2eZdZeeeeegZgdZ e dZ y)FluxAutoBlocks) text_encoder image_encoderbefore_denoiser4decoderc y)NzAuto Modular pipeline for text-to-image and image-to-image using Flux. - for text-to-image generation, all you need to provide is `prompt` - for image-to-image generation, you need to provide either `image` or `image_latents`rrs rrzFluxAutoBlocks.descriptions  g rN) rrrrrr.r3r9r r!r#rrrrr<r<s2! M\K   rr<r=r&r(r'r4decoder>r?)r/rautoN)"utilsrmodular_pipelinerrmodular_pipeline_utilsrr?r r r r r decodersrr4rencodersrr get_loggerrloggerrr%r+r.r3r9r<TEXT2IMAGE_BLOCKSIMAGE2IMAGE_BLOCKS AUTO_BLOCKS ALL_BLOCKSrrrrOswK3%$=   H %  /   4 (  #;   2   ,  g+g - &# ,- - ./ 23 O$ >"  $ ,- ,- - 56 9: O$ >"  ,- 01 45 '( %&  .:LVa b r