i P ddlmZddlmZmZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZmZmZddlmZmZmZmZmZmZmZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)m*Z*ejVe,Z-ed efd efgZ.Gd deZ/ede%fde*fde fdefde fdefd e/fgZ0ede#fde&fgZ1GddeZ2ede*fde)dgdgfgZ3GddeZ4ed e fd!e fgZ5Gd"d#eZ6ed efd efgZ7Gd$d%eZ8ede%fd&e2fde4fde fdefd'e6fde fdefd e8fg Z9ede$fde&fgZ:Gd(d)eZ;ede*fde)dg*fgZ<Gd+d,eZ=ede%fd&e;fde=fde fdefd-e fde fdefd e/fg Z>ed.efd/e(fd0efd1efgZ?Gd2d3eZ@Gd4d5eZAGd6d7eZBGd8d9eZCede fdefde fgZDGd:d;eZEede fdefd'e6fde fgZFGd<d=eZGede fdefd-e fde fgZHGd>d?eZIGd@dAeZJGdBdCeZKGdDdEeZLGdFdGeZMGdHdIeZNGdJdKeZOede%fd&e@fd.eAfdeOfd eNfgZPGdLdMeZQedNe!fde"fgZRGdOdPeZSedNe!fde$fde&fgZTGdQdReZUede*fde)dg*fgZVGdSdTeZWedeSfd&eUfdeWfde fdefde fdefd e/fgZXedNe!fde#fde&dUdVfgZYGdWdXeZZedeSfd&eZfde4fde fdefd'e6fde fdefd e8fg Z[GdYdZeZ\Gd[d\eZ]ede fdefde fgZ^Gd]d^eZ_ede fdefd'e6fde fgZ`Gd_d`eZaGdadbeZbGdcddeZcGdedfeZdedeSfd&e\fdedfd eNfgZeGdgdheZfedNefde fgZgGdidjeZhedNefdefde&fgZiGdkdleZjedehfd&ejfdeWfde fdefde fdefd e/fgZkGdmdneZlGdodpeZmGdqdreZnedehfd&emfdenfd eNfgZoGdsdteZpe0e>eXe[eke9e?ePeeeodu Zqyv)w)logging)AutoPipelineBlocksSequentialPipelineBlocks)InsertableDict)%QwenImageControlNetBeforeDenoiserStepQwenImageCreateMaskLatentsStepQwenImageEditRoPEInputsStepQwenImagePrepareLatentsStep'QwenImagePrepareLatentsWithStrengthStepQwenImageRoPEInputsStepQwenImageSetTimestepsStep%QwenImageSetTimestepsWithStrengthStep)QwenImageDecoderStep'QwenImageInpaintProcessImagesOutputStep QwenImageProcessImagesOutputStep)QwenImageControlNetDenoiseStepQwenImageDenoiseStepQwenImageEditDenoiseStepQwenImageEditInpaintDenoiseStep%QwenImageInpaintControlNetDenoiseStepQwenImageInpaintDenoiseStep%QwenImageLoopBeforeDenoiserControlNet) !QwenImageControlNetVaeEncoderStep'QwenImageEditPlusProcessImagesInputStep"QwenImageEditPlusResizeDynamicStep QwenImageEditPlusTextEncoderStepQwenImageEditResizeDynamicStepQwenImageEditTextEncoderStep&QwenImageInpaintProcessImagesInputStepQwenImageProcessImagesInputStepQwenImageTextEncoderStepQwenImageVaeEncoderDynamicStep)QwenImageControlNetInputsStepQwenImageInputsDynamicStepQwenImageTextInputsStepdecode postprocessc`eZdZdZej ZejZe dZ y)QwenImageDecodeStep qwenimagecy)NzSDecode step that decodes the latents to images and postprocess the generated image.selfs ~/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/modular_pipelines/qwenimage/modular_blocks.py descriptionzQwenImageDecodeStep.descriptionJdN) __name__ __module__ __qualname__ model_nameQwenImageDecodeBlocksvalues block_classeskeys block_namespropertyr2r.r4r1r+r+Es7J)002M',,.K eer4r+ text_encoderinputprepare_latents set_timestepsprepare_rope_inputsdenoise preprocessencodecfeZdZdZej ZejZe de fdZ y)QwenImageInpaintVaeEncoderStepr,returnc y)NzThis step is used for processing image and mask inputs for inpainting tasks. It: - Resizes the image to the target size, based on `height` and `width`. - Processes and updates `image` and `mask_image`. - Creates `image_latents`.r.r/s r1r2z*QwenImageInpaintVaeEncoderStep.descriptionps  * r4N) r5r6r7r8 QwenImageInpaintVaeEncoderBlocksr:r;r<r=r>strr2r.r4r1rHrHks<J4;;=M2779K  S  r4rH text_inputsadditional_inputs image_latentsprocessed_mask_image)image_latent_inputsadditional_batch_inputsc`eZdZdZej ZejZe dZ y)QwenImageInpaintInputStepr,cy)NzKInput step that prepares the inputs for the inpainting denoising step. It: r.r/s r1r2z%QwenImageInpaintInputStep.descriptions]r4N) r5r6r7r8QwenImageInpaintInputBlocksr:r;r<r=r>r2r.r4r1rTrT7J/668M-224K RRr4rTadd_noise_to_latentscreate_mask_latentscfeZdZdZej ZejZe de fdZ y)"QwenImageInpaintPrepareLatentsStepr,rIc y)NzThis step prepares the latents/image_latents and mask inputs for the inpainting denoising step. It: - Add noise to the image latents to create the latents input for the denoiser. - Create the pachified latents `mask` based on the processedmask image. r.r/s r1r2z.QwenImageInpaintPrepareLatentsStep.descriptions  Y r4N) r5r6r7r8$QwenImageInpaintPrepareLatentsBlocksr:r;r<r=r>rLr2r.r4r1r[r[s<J8??AM6;;=K  S  r4r[c`eZdZdZej ZejZe dZ y)QwenImageInpaintDecodeStepr,cy)NzDecode step that decodes the latents to images and postprocess the generated image, optional apply the mask overally to the original image.r.r/s r1r2z&QwenImageInpaintDecodeStep.descriptions]r4N) r5r6r7r8QwenImageInpaintDecodeBlocksr:r;r<r=r>r2r.r4r1r_r_s7J0779M.335K ]]r4r_ vae_encoderprepare_inpaint_latentscfeZdZdZej ZejZe de fdZ y)QwenImageImg2ImgVaeEncoderStepr,rIcy)Nz^Vae encoder step that preprocess andencode the image inputs into their latent representations.r.r/s r1r2z*QwenImageImg2ImgVaeEncoderStep.descriptionsor4N) r5r6r7r8 QwenImageImg2ImgVaeEncoderBlocksr:r;r<r=r>rLr2r.r4r1reres?J4;;=M2779K pSppr4re)rQc`eZdZdZej ZejZe dZ y)QwenImageImg2ImgInputStepr,cy)NzHInput step that prepares the inputs for the img2img denoising step. It: r.r/s r1r2z%QwenImageImg2ImgInputStep.descriptionsZr4N) r5r6r7r8QwenImageImg2ImgInputBlocksr:r;r<r=r>r2r.r4r1ririrWr4riprepare_img2img_latentscontrolnet_vae_encodercontrolnet_inputscontrolnet_before_denoisecontrolnet_denoise_loop_beforec4eZdZeegZddgZddgZedZ y)QwenImageAutoVaeEncoderStepinpaintimg2img mask_imageimagec y)NaZVae encoder step that encode the image inputs into their latent representations. This is an auto pipeline block. - `QwenImageInpaintVaeEncoderStep` (inpaint) is used when `mask_image` is provided. - `QwenImageImg2ImgVaeEncoderStep` (img2img) is used when `image` is provided. - if `mask_image` or `image` is not provided, step will be skipped.r.r/s r1r2z'QwenImageAutoVaeEncoderStep.description%  U r4N) r5r6r7rHrer;r=block_trigger_inputsr>r2r.r4r1rrrr s435STMi(K('2   r4rrc.eZdZegZdgZdgZedZy))QwenImageOptionalControlNetVaeEncoderStep controlnet control_imagec y)Na Vae encoder step that encode the image inputs into their latent representations. This is an auto pipeline block. - `QwenImageControlNetVaeEncoderStep` (controlnet) is used when `control_image` is provided. - if `control_image` is not provided, step will be skipped.r.r/s r1r2z5QwenImageOptionalControlNetVaeEncoderStep.description6  M r4N) r5r6r7rr;r=ryr>r2r.r4r1r{r{1s,67M.K+,   r4r{c6eZdZeeegZgdZgdZe dZ y)QwenImageAutoInputSteprsrt text2imagerPrONc y)NaInput step that standardize the inputs for the denoising step, e.g. make sure inputs have consistent batch size, and patchified. This is an auto pipeline block that works for text2image/inpaint/img2img tasks. - `QwenImageInpaintInputStep` (inpaint) is used when `processed_mask_image` is provided. - `QwenImageImg2ImgInputStep` (img2img) is used when `image_latents` is provided. - `QwenImageTextInputsStep` (text2image) is used when both `processed_mask_image` and `image_latents` are not provided. r.r/s r1r2z"QwenImageAutoInputStep.descriptionIs  K r4N) r5r6r7rTrir'r;r=ryr>r2r.r4r1rrDs-.0IKbcM6KJ   r4rc.eZdZegZdgZdgZedZy)$QwenImageOptionalControlNetInputStepr|control_image_latentsc y)Na Controlnet input step that prepare the control_image_latents input. This is an auto pipeline block. - `QwenImageControlNetInputsStep` (controlnet) is used when `control_image_latents` is provided. - if `control_image_latents` is not provided, step will be skipped.r.r/s r1r2z0QwenImageOptionalControlNetInputStep.descriptionZ  U r4N) r5r6r7r%r;r=ryr>r2r.r4r1rrUs,23M.K34   r4rc`eZdZdZej ZejZe dZ y)$QwenImageText2ImageBeforeDenoiseStepr,cy)Nz|Before denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step for text2image task.r.r/s r1r2z0QwenImageText2ImageBeforeDenoiseStep.descriptionvsNr4N) r5r6r7r8&QwenImageText2ImageBeforeDenoiseBlocksr:r;r<r=r>r2r.r4r1rrqs7J:AACM8==?K NNr4rc`eZdZdZej ZejZe dZ y)!QwenImageInpaintBeforeDenoiseStepr,cy)NzyBefore denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step for inpaint task.r.r/s r1r2z-QwenImageInpaintBeforeDenoiseStep.descriptionKr4N) r5r6r7r8#QwenImageInpaintBeforeDenoiseBlocksr:r;r<r=r>r2r.r4r1rr7J7>>@M5::r2r.r4r1rrrr4rc6eZdZeeegZgdZgdZe dZ y)QwenImageAutoBeforeDenoiseSteprrc y)NaBefore denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step. This is an auto pipeline block that works for text2img, inpainting, img2img tasks. - `QwenImageInpaintBeforeDenoiseStep` (inpaint) is used when `processed_mask_image` is provided. - `QwenImageImg2ImgBeforeDenoiseStep` (img2img) is used when `image_latents` is provided. - `QwenImageText2ImageBeforeDenoiseStep` (text2image) is used when both `processed_mask_image` and `image_latents` are not provided. r.r/s r1r2z*QwenImageAutoBeforeDenoiseStep.descriptions  X r4N) r5r6r7rrrr;r=ryr>r2r.r4r1rrs1)),M 7KJ   r4rc.eZdZegZdgZdgZedZy),QwenImageOptionalControlNetBeforeDenoiseStepr|rc y)NaControlnet before denoise step that prepare the controlnet input. This is an auto pipeline block. - `QwenImageControlNetBeforeDenoiserStep` (controlnet) is used when `control_image_latents` is provided. - if `control_image_latents` is not provided, step will be skipped.r.r/s r1r2z8QwenImageOptionalControlNetBeforeDenoiseStep.descriptionrr4N) r5r6r7r r;r=ryr>r2r.r4r1rrs,:;M.K34   r4rc4eZdZeegZddgZddgZedZ y)"QwenImageControlNetAutoDenoiseStepinpaint_denoiserDmaskNc y)NaHControlnet step during the denoising process. This is an auto pipeline block that works for inpaint and text2image/img2img tasks with controlnet. - `QwenImageInpaintControlNetDenoiseStep` (inpaint) is used when `mask` is provided. - `QwenImageControlNetDenoiseStep` (text2image/img2img) is used when `mask` is not provided. r.r/s r1r2z.QwenImageControlNetAutoDenoiseStep.descriptions  p r4) r5r6r7rrr;r=ryr>r2r.r4r1rrs3:   r4rc6eZdZeeegZgdZgdZe dZ y)QwenImageAutoDenoiseStep)controlnet_denoiserrD)rrNc y)NaDenoise step that iteratively denoise the latents. This is an auto pipeline block that works for inpaint/text2image/img2img tasks. It also works with controlnet - `QwenImageControlNetAutoDenoiseStep` (controlnet) is used when `control_image_latents` is provided. - `QwenImageInpaintDenoiseStep` (inpaint) is used when `mask` is provided and `control_image_latents` is not provided. - `QwenImageDenoiseStep` (text2image/img2img) is used when `mask` is not provided and `control_image_latents` is not provided. r.r/s r1r2z$QwenImageAutoDenoiseStep.descriptions  R r4N) r5r6r7rrrr;r=ryr>r2r.r4r1rrs2*#M GKB   r4rc4eZdZeegZddgZddgZedZ y)QwenImageAutoDecodeStepinpaint_decoder(rNc y)NaHDecode step that decode the latents into images. This is an auto pipeline block that works for inpaint/text2image/img2img tasks, for both QwenImage and QwenImage-Edit. - `QwenImageInpaintDecodeStep` (inpaint) is used when `mask` is provided. - `QwenImageDecodeStep` (text2image/img2img) is used when `mask` is not provided. r.r/s r1r2z#QwenImageAutoDecodeStep.descriptions  e r4) r5r6r7r_r+r;r=ryr>r2r.r4r1rrs3/1DEM#X.K"D>   r4rc6eZdZdZeeeeegZ gdZ e dZ y)QwenImageCoreDenoiseStepr,)r@controlnet_inputbefore_denoiserorDc y)NaCore step that performs the denoising process. - `QwenImageAutoInputStep` (input) standardizes the inputs for the denoising step. - `QwenImageOptionalControlNetInputStep` (controlnet_input) prepares the controlnet input. - `QwenImageAutoBeforeDenoiseStep` (before_denoise) prepares the inputs for the denoising step. - `QwenImageOptionalControlNetBeforeDenoiseStep` (controlnet_before_denoise) prepares the controlnet input for the denoising step. - `QwenImageAutoDenoiseStep` (denoise) iteratively denoises the latents. This step support text-to-image, image-to-image, inpainting, and controlnet tasks for QwenImage: - for image-to-image generation, you need to provide `image_latents` - for inpainting, you need to provide `processed_mask_image` and `image_latents` - to run the controlnet workflow, you need to provide `control_image_latents` - for text-to-image generation, all you need to provide is prompt embeddingsr.r/s r1r2z$QwenImageCoreDenoiseStep.descriptions  ^ r4N) r5r6r7r8rrrrrr;r=r>r2r.r4r1rrs7J,&4 MjK     r4rc`eZdZdZej ZejZe dZ y)QwenImageAutoBlocksr,c y)NaAuto Modular pipeline for text-to-image, image-to-image, inpainting, and controlnet tasks using QwenImage. - for image-to-image generation, you need to provide `image` - for inpainting, you need to provide `mask_image` and `image`, optionally you can provide `padding_mask_crop` - to run the controlnet workflow, you need to provide `control_image` - for text-to-image generation, all you need to provide is `prompt`r.r/s r1r2zQwenImageAutoBlocks.description3s  T r4N) r5r6r7r8 AUTO_BLOCKSr:r;r<r=r>r2r.r4r1rr-s5J&&(M""$K   r4rresizecfeZdZdZej ZejZe de fdZ y)QwenImageEditVLEncoderStepr,rIcy)NzNQwenImage-Edit VL encoder step that encode the image an text prompts together.r.r/s r1r2z&QwenImageEditVLEncoderStep.descriptionPs_r4N) r5r6r7r8QwenImageEditVLEncoderBlocksr:r;r<r=r>rLr2r.r4r1rrKs?J0779M.335K `S``r4rcfeZdZdZej ZejZe de fdZ y)QwenImageEditVaeEncoderStepr,rIcyNzPVae encoder step that encode the image inputs into their latent representations.r.r/s r1r2z'QwenImageEditVaeEncoderStep.descriptiondar4N) r5r6r7r8QwenImageEditVaeEncoderBlocksr:r;r<r=r>rLr2r.r4r1rr_s?J188:M/446K bSbbr4rc`eZdZdZej ZejZe dZ y)QwenImageEditInputStepr,cy)NzEInput step that prepares the inputs for the edit denoising step. It: r.r/s r1r2z"QwenImageEditInputStep.descriptionwsWr4N) r5r6r7r8QwenImageEditInputBlocksr:r;r<r=r>r2r.r4r1rrrs7J,335M*//1K RRr4rprocessed_image) input_name output_namecfeZdZdZej ZejZe de fdZ y)"QwenImageEditInpaintVaeEncoderStepr,rIc y)NzThis step is used for processing image and mask inputs for QwenImage-Edit inpaint tasks. It: - resize the image for target area (1024 * 1024) while maintaining the aspect ratio. - process the resized image and mask image. - create image latents.r.r/s r1r2z.QwenImageEditInpaintVaeEncoderStep.descriptions  ' r4N) r5r6r7r8$QwenImageEditInpaintVaeEncoderBlocksr:r;r<r=r>rLr2r.r4r1rrs<J8??AM6;;=K  S  r4rc4eZdZeegZddgZddgZedZ y)QwenImageEditAutoVaeEncoderStep edit_inpainteditrurvc y)NaVae encoder step that encode the image inputs into their latent representations. This is an auto pipeline block that works for edit and edit_inpaint tasks. - `QwenImageEditInpaintVaeEncoderStep` (edit_inpaint) is used when `mask_image` is provided. - `QwenImageEditVaeEncoderStep` (edit) is used when `image` is provided. - if `mask_image` or `image` is not provided, step will be skipped.r.r/s r1r2z+QwenImageEditAutoVaeEncoderStep.descriptionrxr4N) r5r6r7rrr;r=ryr>r2r.r4r1rrs8*#M"6*K('2   r4rc4eZdZeegZddgZddgZedZ y)QwenImageEditAutoInputSteprrrPrOc y)NaInput step that prepares the inputs for the edit denoising step. It is an auto pipeline block that works for edit and edit_inpaint tasks. - `QwenImageInpaintInputStep` (edit_inpaint) is used when `processed_mask_image` is provided. - `QwenImageEditInputStep` (edit) is used when `image_latents` is provided. - if `processed_mask_image` or `image_latents` is not provided, step will be skipped.r.r/s r1r2z&QwenImageEditAutoInputStep.description  g r4N) r5r6r7rTrr;r=ryr>r2r.r4r1rrs4.0FGM!6*K2OD   r4rc`eZdZdZej ZejZe dZ y)QwenImageEditBeforeDenoiseStepr,cy)NzvBefore denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step for edit task.r.r/s r1r2z*QwenImageEditBeforeDenoiseStep.descriptionsHr4N) r5r6r7r8 QwenImageEditBeforeDenoiseBlocksr:r;r<r=r>r2r.r4r1rrs7J4;;=M2779K HHr4rc`eZdZdZej ZejZe dZ y)%QwenImageEditInpaintBeforeDenoiseStepr,cy)Nz~Before denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step for edit inpaint task.r.r/s r1r2z1QwenImageEditInpaintBeforeDenoiseStep.description sPr4N) r5r6r7r8'QwenImageEditInpaintBeforeDenoiseBlocksr:r;r<r=r>r2r.r4r1rrs7J;BBDM9>>@K PPr4rc8eZdZdZeegZddgZddgZe dZ y)"QwenImageEditAutoBeforeDenoiseStepqwenimage-editrrrPrOc y)NaBefore denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step. This is an auto pipeline block that works for edit (img2img) and edit inpaint tasks. - `QwenImageEditInpaintBeforeDenoiseStep` (edit_inpaint) is used when `processed_mask_image` is provided. - `QwenImageEditBeforeDenoiseStep` (edit) is used when `image_latents` is provided and `processed_mask_image` is not provided. - if `image_latents` or `processed_mask_image` is not provided, step will be skipped.r.r/s r1r2z.QwenImageEditAutoBeforeDenoiseStep.descriptionrr4N) r5r6r7r8rrr;r=ryr>r2r.r4r1rrs=!J-&M"6*K2OD   r4rc8eZdZdZeegZddgZddgZe dZ y)QwenImageEditAutoDenoiseSteprrrDrPrOc y)NaDenoise step that iteratively denoise the latents. This block supports edit (img2img) and edit inpaint tasks for QwenImage Edit. - `QwenImageEditInpaintDenoiseStep` (inpaint) is used when `processed_mask_image` is provided. - `QwenImageEditDenoiseStep` (img2img) is used when `image_latents` is provided. - if `processed_mask_image` or `image_latents` is not provided, step will be skipped.r.r/s r1r2z(QwenImageEditAutoDenoiseStep.description1rr4N) r5r6r7r8rrr;r=ryr>r2r.r4r1rr*s9!J46NOM$i0K2OD   r4rc2eZdZdZeeegZgdZe dZ y)QwenImageEditCoreDenoiseSteprr@rrDc y)Na2Core step that performs the denoising process. - `QwenImageEditAutoInputStep` (input) standardizes the inputs for the denoising step. - `QwenImageEditAutoBeforeDenoiseStep` (before_denoise) prepares the inputs for the denoising step. - `QwenImageEditAutoDenoiseStep` (denoise) iteratively denoises the latents. This step support edit (img2img) and edit inpainting workflow for QwenImage Edit: - When `processed_mask_image` is provided, it will be used for edit inpainting task. - When `image_latents` is provided, it will be used for edit (img2img) task. r.r/s r1r2z(QwenImageEditCoreDenoiseStep.descriptionHs  ` r4N) r5r6r7r8rrrr;r=r>r2r.r4r1rr?s0!J"*$M 9K     r4rc`eZdZdZej ZejZe dZ y)QwenImageEditAutoBlocksrc y)NaAuto Modular pipeline for edit (img2img) and edit inpaint tasks using QwenImage-Edit. - for edit (img2img) generation, you need to provide `image` - for edit inpainting, you need to provide `mask_image` and `image`, optionally you can provide `padding_mask_crop` r.r/s r1r2z#QwenImageEditAutoBlocks.descriptionds  G r4N) r5r6r7r8EDIT_AUTO_BLOCKSr:r;r<r=r>r2r.r4r1rr_s5!J$++-M"'')K   r4rcfeZdZdZej ZejZe de fdZ y)QwenImageEditPlusVLEncoderStepr,rIcy)NzSQwenImage-Edit Plus VL encoder step that encode the image an text prompts together.r.r/s r1r2z*QwenImageEditPlusVLEncoderStep.descriptionr3r4N) r5r6r7r8 QwenImageEditPlusVLEncoderBlocksr:r;r<r=r>rLr2r.r4r1rr|s?J4;;=M2779K eSeer4rcfeZdZdZej ZejZe de fdZ y)QwenImageEditPlusVaeEncoderStepr,rIcyrr.r/s r1r2z+QwenImageEditPlusVaeEncoderStep.descriptionrr4N) r5r6r7r8!QwenImageEditPlusVaeEncoderBlocksr:r;r<r=r>rLr2r.r4r1rrs?J5<<>M388:K bSbbr4rc2eZdZdZegZdgZdgZedZ y)&QwenImageEditPlusAutoBeforeDenoiseStepqwenimage-edit-plusrrOc y)NahBefore denoise step that prepare the inputs (timesteps, latents, rope inputs etc.) for the denoise step. This is an auto pipeline block that works for edit (img2img) task. - `QwenImageEditBeforeDenoiseStep` (edit) is used when `image_latents` is provided and `processed_mask_image` is not provided. - if `image_latents` is not provided, step will be skipped.r.r/s r1r2z2QwenImageEditPlusAutoBeforeDenoiseStep.descriptionrr4N) r5r6r7r8rr;r=ryr>r2r.r4r1rrs1&J34M(K+,   r4rc.eZdZegZdgZdgZedZy)#QwenImageEditPlusAutoVaeEncoderSteprrvc y)NaVae encoder step that encode the image inputs into their latent representations. This is an auto pipeline block that works for edit task. - `QwenImageEditPlusVaeEncoderStep` (edit) is used when `image` is provided. - if `image` is not provided, step will be skipped.r.r/s r1r2z/QwenImageEditPlusAutoVaeEncoderStep.descriptions  E r4N) r5r6r7rr;r=ryr>r2r.r4r1rrs/'M(K#9   r4rc2eZdZdZeeegZgdZe dZ y) QwenImageEditPlusCoreDenoiseSteprrc y)NaCore step that performs the denoising process. - `QwenImageEditAutoInputStep` (input) standardizes the inputs for the denoising step. - `QwenImageEditPlusAutoBeforeDenoiseStep` (before_denoise) prepares the inputs for the denoising step. - `QwenImageEditAutoDenoiseStep` (denoise) iteratively denoises the latents. This step support edit (img2img) workflow for QwenImage Edit Plus: - When `image_latents` is provided, it will be used for edit (img2img) task. r.r/s r1r2z,QwenImageEditPlusCoreDenoiseStep.descriptions  ` r4N) r5r6r7r8rrrr;r=r>r2r.r4r1rrs0&J".$M 9K   r4rc`eZdZdZej ZejZe dZ y)QwenImageEditPlusAutoBlocksrc y)NzAuto Modular pipeline for edit (img2img) and edit tasks using QwenImage-Edit Plus. - for edit (img2img) generation, you need to provide `image` r.r/s r1r2z'QwenImageEditPlusAutoBlocks.descriptions  O r4N) r5r6r7r8EDIT_PLUS_AUTO_BLOCKSr:r;r<r=r>r2r.r4r1rrs5&J)002M',,.K   r4r) rrtrr edit_plusrsr|auto edit_autoedit_plus_autoN)rutilsrmodular_pipelinerrmodular_pipeline_utilsrrr r r r r rrrdecodersrrrrDrrrrrrrencodersrrrrrr r!r"r#r$inputsr%r&r' get_loggerr5loggerr9r+TEXT2IMAGE_BLOCKSrKrHrVrTr]r[rar_INPAINT_BLOCKSrgrerkriIMAGE2IMAGE_BLOCKSCONTROLNET_BLOCKSrrr{rrrrrrrrrrrrrrrrrrrrrr EDIT_BLOCKSrrEDIT_INPAINT_BLOCKSrrrrrrrrrrrrrrrEDIT_PLUS_BLOCKSrrrrr ALL_BLOCKSr.r4r1rs K3   vu   gf   H %' ')* 8:;e2e# 134 )+, 79: 356  7 9: (*+ &()  $2  2  134 $   %=  - /12  &%4$5PfOg    R 8 R(6 !H!JK  > @A($  )A   . ')* ?AB ]!9] 134 689 +-. 79: ?AB "$F$HI  7 9: /12 -/0  $$2 689 134$ p%=p- /12 8oM^_` R 8 R$ 134 689 +-. 79: ?AB "$K$MN  7 9: (*+ &()  $# !#D#FG ;=> ' 1 3 - 1 3   (  "4  "  0B  &  /  "  +=  &*8 79: 356  7 9:*&N+CN'5 79: ?AB "$F$HI  7 9: '#K(@K'5 79: ?AB "$K$MN  7 9: '#K(@K %7 *  3E  &  );   1 0  0   7 : 134 356 !#L#NO ,./ *,-   2 , . 134 /12 `!9`!/ 134 689 134!b":b* /12 8oM^_` R5 R 356 356 (*+ 79: 356  ; => ,./ &()   "(6 134  2  *6GUd e  ($  )A  % 356 :<= +-. 79: ?AB "$F$HI  ; => 356 -/0  $ &8 (  !3  ($2 79: 356  ; =>$ H%=H+9 79: ?AB "$F$HI  ; => +'P,DP ); . #5 * #; ," 356 79: 023 *,-   6  *$2 578 356$ e%=e%3 578 >@A 134%!b&>b" 79: 79: (*+ 79: 356  ; => ,./ &()    -?  & *< ( '? *' 79: ;=> 467 *,-   ":  "$! '!# !+  r4