+h& SSKJr SSKJrJr SSKJr SSKJrJ r J r J r J r SSK Jr SSKJr SS KJrJr \R("\5r"S S \5r"S S \5r"SS\5r"SS\5r"SS\5r"SS\5r"SS\5r\"S\4S\ 4S\ 4S\ 4S\4S\4/5r\"S\4S\4S\ 4S\ 4S\4S\4S\4/5r\"S\4S\4S\4S\4S\4/5r \\\ S .r!g!)")logging)AutoPipelineBlocksSequentialPipelineBlocks)InsertableDict)FluxImg2ImgPrepareLatentsStepFluxImg2ImgSetTimestepsStep FluxInputStepFluxPrepareLatentsStepFluxSetTimestepsStep)FluxDecodeStep)FluxDenoiseStep)FluxTextEncoderStepFluxVaeEncoderStepc6\rSrSr\/rS/rS/r\S5r Sr g)FluxAutoVaeEncoderStep"img2imgimagecg)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 i/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/modular_pipelines/flux/modular_blocks.py description"FluxAutoVaeEncoderStep.description's  A rN) __name__ __module__ __qualname____firstlineno__r block_classes block_namesblock_trigger_inputspropertyr__static_attributes__rrrrr"s+'(M+K#9   rrc6\rSrSr\\\/r/SQr\ S5r Sr g)FluxBeforeDenoiseStep2)inputprepare_latents set_timestepscg)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 rr!FluxBeforeDenoiseStep.description:s  I rrN) rr r!r"r r r r#r$r&rr'rrrr)r)2s,M @K   rr)c6\rSrSr\\\/r/SQr\ S5r Sr g)FluxImg2ImgBeforeDenoiseStepF)r+r-r,cg)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 rr(FluxImg2ImgBeforeDenoiseStep.descriptionJs  T rrN) rr r!r"r r r r#r$r&rr'rrrr1r1Fs'"$?A^_M?K   rr1c<\rSrSr\\/rSS/rSS/r\ S5r Sr g)FluxAutoBeforeDenoiseStepV text2imagerN image_latentscg)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 rr%FluxAutoBeforeDenoiseStep.description[s  m rr) rr r!r"r)r1r#r$r%r&rr'rrrr6r6Vs4*,HIM+K /2   rr6c>\rSrSr\/rS/rS/r\S\ 4Sj5r Sr g)FluxAutoDenoiseStepfdenoiseNreturncg)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 rrFluxAutoDenoiseStep.descriptionks  O rr) rr r!r"rr#r$r%r&strrr'rrrr=r=fs2$%M+K 6  S  rr=c6\rSrSr\/rS/rS/r\S5r Sr g)FluxAutoDecodeStepuz non-inpaintNcg)NzTDecode step that decode the denoised latents into image outputs. - `FluxDecodeStep`rrs rrFluxAutoDecodeStep.descriptionzsfrr) rr r!r"rr#r$r%r&rr'rrrrErEus-#$M /K 6 ggrrEc:\rSrSr\\\\\/r /SQr \ S5r Sr g)FluxAutoBlocks) text_encoder image_encoderbefore_denoiser?decodercg)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 rrFluxAutoBlocks.descriptions  g rrN)rr r!r"rrr6r=rEr#r$r&rr'rrrrJrJs2! M\K   rrJrLr+r-r,r?decoderMrN)r8rautoN)"utilsrmodular_pipelinerrmodular_pipeline_utilsrrNr r r r r decodersrr?rencodersrr get_loggerrloggerrr)r1r6r=rErJTEXT2IMAGE_BLOCKSIMAGE2IMAGE_BLOCKS AUTO_BLOCKS ALL_BLOCKSrrrr_suK3%$=   H %  /   4 (  #;   2   ,  g+g - &# ,- - ./ 23 O$ >"  $ ,- ,- - 56 9: O$ >"  ,- 01 45 '( %&  .:LVa b r