ifddlmZmZmZmZddlZddlZddlmZm Z m Z ddl m Z ddl mZddlmZmZmZmZddlmZmZmZdd lmZdd lmZdd lmZd d lmZm Z d dl!m"Z"m#Z#m$Z$m%Z%ddlm&Z&ejNe(Z)dejTdejTfdZ+ d:dee,ee,fde,de-de-deej\f dZ/ d;dee,ee,fdeejTde,de-deej\f dZ0 ddejTd"edejfdej\d#ejjd$e-d e,fd%Z6Gd&d'eZ7Gd(d)e7Z8Gd*d+eZ9Gd,d-eZ:Gd.d/e:Z;Gd0d1eZ<Gd2d3eZ=Gd4d5e=Z>Gd6d7eZ?Gd8d9eZ@y)?)DictListOptionalUnionN)"Qwen2_5_VLForConditionalGenerationQwen2TokenizerQwen2VLProcessor) FrozenDict)ClassifierFreeGuidance)InpaintProcessorVaeImageProcessoris_valid_imageis_valid_image_imagelist)AutoencoderKLQwenImageQwenImageControlNetModelQwenImageMultiControlNetModel)calculate_dimensions)logging) unwrap_module)ModularPipelineBlocks PipelineState) ComponentSpec ConfigSpec InputParam OutputParam)QwenImageModularPipeline hidden_statesmaskc|j}|jd}||}tj||j d}|S)Nrdimr)boolsumtorchsplittolist)r r! bool_mask valid_lengthsselected split_results x/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/modular_pipelines/qwenimage/encoders.py_extract_masked_hiddenr/$sH IMMaM(MY'H;;x)=)=)?QGL promptprompt_template_encode prompt_template_encode_start_idxtokenizer_max_lengthdevicec.t|tr|gn|}|}|}|D cgc]} |j| } } || ||zdddj|} || j| j d} | j d} t| | j }|D cgc]} | |d }} |D cgc]A} tj| jdtj| jC}} t|D cgc]} | jdc} }tj|Dcgc]J}tj||j!||jdz |jdgLc}}tj|Dcgc]:}tj||j!||jdz g<c}}|j| }||fScc} wcc} wcc} wcc} wcc}wcc}w) NTpt) max_lengthpadding truncationreturn_tensors) input_idsattention_maskoutput_hidden_statesrdtyper5rr5) isinstancestrformattor<r=r r/r'onessizelongr5maxstackcat new_zeros) text_encoder tokenizerr1r2r3r4r5templatedrop_idxetxt txt_tokensencoder_hidden_statesr split_hidden_statesattn_mask_list max_seq_lenu prompt_embedsencoder_attention_masks r.get_qwen_prompt_embedsr\,s$FC0fXfF%H/H'- .!8??1  .C . ,x7RVgkbj)&&!00! *77;M0 @Y@YZ1DEA1XY<EEXklSTejj%**QXXNlNl*=>Qqvvay>?KKKRefQAq{{;#:AFF1IFG HfM#[[GUV!Aq{{;#:;< =V"$$F$3M 0 001 /Fl>f Ws%G9 G>1AHH4AH ?HimagecRt|tr|gn|}|}|}|D cgc]} |j| } } || |ddj|} || j| j | j | jd} | jd} t| | j }|D cgc]} | |d }} |D cgc]A} tj| jdtj| jC}} t|D cgc]} | jdc} }tj |Dcgc]J}tj"||j%||jdz |jdgLc}}tj |Dcgc]:}tj"||j%||jdz g<c}}|j| }||fScc} wcc} wcc} wcc} wcc}wcc}w) NTr7textimagesr9r;r<r= pixel_valuesimage_grid_thwr>r?rr@rrB)rCrDrErFr<r=rcrdr r/r'rGrHrIr5rJrKrLrM)rN processorr1r]r2r3r5rPrQrRrS model_inputsoutputsr rVrWrXrYrZr[s r.get_qwen_prompt_embeds_editrhTs$FC0fXfF%H/H'- .!8??1  .C .    bj ((#22!..#22! G))"-M0 @[@[\1DEA1XY<EEXklSTejj%**QXXNlNl*=>Qqvvay>?KKKRefQAq{{;#:AFF1IFG HfM#[[GUV!Aq{{;#:;< =V"$$F$3M 0 00? /&Fl>f Ws%H 1 HAHHAH/?H$img_template_encodect|tr|gn|}t|tr-d}t|D]\} } ||j | dzz }n||j d}nd}|} |} |D cgc]} | j || z}} |||ddj |}||j |j|j|jd}|jd}t||j}|D cgc]} | | d }} |D cgc]A} tj| jdtj| j C}} t#|D cgc]} | jdc} }tj$|Dcgc]J}tj&||j)||jdz |jdgLc}}tj$|Dcgc]:}tj&||j)||jdz g<c}}|j | }||fScc} wcc} wcc} wcc} wcc}wcc}w) NrTr7r_rbr?rr@rB)rCrDlist enumeraterErFr<r=rcrdr r/r'rGrHrIr5rJrKrLrM)rNrer1r]r2rir3r5base_img_promptiimgrPrQrRrSrfrgr rVrWrXrYrZr[s r. get_qwen_prompt_embeds_edit_plusrqsM$FC0fXfF%& AFAs 299!a%@ @O A  -44Q7%H/H9? @A8???Q. / @C @    bj  ((#22!..#22! G))"-M0 @[@[\1DEA1XY<EEXklSTejj%**QXXNlNl*=>Qqvvay>?KKKRefQAq{{;#:AFF1IFG HfM#[[GUV!Aq{{;#:;< =V"$$F$3M 0 00; A$Fl>f Ws&1I! I&AI+*I0AI5?I:encoder_output generator sample_modect|dr |dk(r|jj|St|dr|dk(r|jjSt|dr |jSt d)N latent_distsampleargmaxlatentsz3Could not access latents of provided encoder_output)hasattrrvrwmoderyAttributeError)rrrsrts r.retrieve_latentsr}st~}-+2I))00;;  /K84K))..00  +%%%RSSr0vaerAlatent_channelsc t|tjstdt |d|j dk(r|j d}n0|j dk7rtd|j d|j||}t|trat|jdDcgc](}t|j|||d z||| *}}tj|d }nt|j||| }tj|jj j#d |d d d j|j$|j&} tj|jj(j#d |d d d j|j$|j&} || z | z }|Scc}w) Nz#Expected image to be a tensor, got .rz Expected image dims 4 or 5, got )r5rArr)rsrtr#)rCr'Tensor ValueErrortyper$ unsqueezerFrlrangeshaper}encoderLtensorconfig latents_meanviewr5rA latents_std) r]r~rsr5rArrtro image_latentsrrs r.encode_vae_imagers eU\\ *>tE{m1MNN yy{a"  ;EIIK=JKK HHF%H 0E)T"5;;q>*  SZZa!a%(89Yq\_j k   -Q7 (E):i]hi  SZZ,,- a!Q * M -"5"5 6  SZZ++, a!Q * M -"5"5 6 #\1[@M ' s-G3ceZdZdZd dedeffd ZedefdZedee fdZ edee fdZ edee fd Zej d ed efd ZxZS)QwenImageEditResizeDynamicStep qwenimage input_name output_namect|trt|ts#tdt|dt|||_||_t |y)aCreate a configurable step for resizing images to the target area (1024 * 1024) while maintaining the aspect ratio. This block resizes an input image tensor and exposes the resized result under configurable input and output names. Use this when you need to wire the resize step to different image fields (e.g., "image", "control_image") Args: input_name (str, optional): Name of the image field to read from the pipeline state. Defaults to "image". output_name (str, optional): Name of the resized image field to write back to the pipeline state. Defaults to "resized_image". 3input_name and output_name must be strings but are  and N)rCrDrr_image_input_name_resized_image_output_namesuper__init__selfrr __class__s r.rz'QwenImageEditResizeDynamicStep.__init__sa*c**[#2NEd:FVEWW\]abm]n\op ",*5' r0returnc"d|jdS)Nz"Image Resize step that resize the zE to the target area (1024 * 1024) while maintaining the aspect ratio.)rrs r. descriptionz*QwenImageEditResizeDynamicStep.description s#3D4J4J3KLQR Rr0c@tdttddidgS)Nimage_resize_processorvae_scale_factor from_configrdefault_creation_methodrrr rs r.expected_componentsz2QwenImageEditResizeDynamicStep.expected_componentss- (!!#5r":;(5    r0cRt|jdtjdgS)NTzThe image to resizenamerequired type_hintr)rrr'rrs r.inputsz%QwenImageEditResizeDynamicStep.inputss* ++dell`u   r0crt|jttjjdgS)NzThe resized imagesrrr)rrrPILImagers r.intermediate_outputsz3QwenImageEditResizeDynamicStep.intermediate_outputs#s1 44SYY__@Ucw   r0 componentsstatec|j|}t||j}t|st dt |t |r|g}|dj\}}td||z \}}} |D cgc] } |jj| ||"} } t||j| |j||||fScc} w)N/Images must be image or list of images but are r)heightwidth)get_block_stategetattrrrrrrrHrrresizesetattrrset_block_state) rrr block_statera image_width image_heightcalculated_widthcalculated_height_r]resized_imagess r.__call__z'QwenImageEditResizeDynamicStep.__call__+s**51 d&<&<='/NtTZ|n]^ ^ & !XF$*1INN! \1EkS^amSm1n.+Q    - - 4 4UCT\l 4 m   T<w+ 00syy/g /   r0rrc|j|}t||j}t|st dt |t |tjs7t |tjjrt |ts|g}g}g}|D]k}|j\}} t|j|| z \} } } |j|j j#|| | |j|mt%||j&|t%||j(||j+||||fS)Nr)rrrrrrrCr'rrrrlrHrrappendrrrrrr) rrrrracondition_images vae_imagesrprrcondition_widthcondition_heightrs r.rz+QwenImageEditPlusResizeDynamicStep.__call__ps:**51 d&<&<='/NtTZ|n]^ ^65<<06399??3vt,XF  #C(+ %K3G));+E4 0O-q  # #J$E$E$L$LSRbds$t u   c "  #  T<<>NO T88*E UK05  r0)r]r vae_image)rrrrrDrrrrrr'rrrrrrs@r.rrDsJ"*%0  # > d;&7  U]]_!#;!M!!r0rceZdZdZedefdZedeefdZ edee fdZ edee fdZ edeefdZedZej&d ed efd Zy ) QwenImageTextEncoderSteprrcy)NzMText Encoder step that generate text_embeddings to guide the image generationrs r.rz$QwenImageTextEncoderStep.descriptions^r0c tdtdtdtdtdtt ddid gS) NrNzThe text encoder to use)rrOzThe tokenizer to useguiderguidance_scale@rr)rrrr r rs r.rz,QwenImageTextEncoderStep.expected_componentssH .*LZs t +~CY Z &!#3S"9:(5    r0cNtddtddtddgS)Nr2<|im_start|>system Describe the image by detailing the color, shape, size, texture, quantity, text, spatial relationships of the objects and background:<|im_end|> <|im_start|>user {}<|im_end|> <|im_start|>assistant rdefaultr3"r4rrs r.expected_configsz)QwenImageTextEncoderStep.expected_configss7 -w  > K 2D A   r0cptddtdtdtdtdtd d gS) Nr1TThe prompt to encodernegative_promptThe negative prompt to encodermax_sequence_lengthzThe max sequence length to user)rrrr)rrDintrs r.rzQwenImageTextEncoderStep.inputss> HtsPf g -Ji j *cGgqu   r0c tddtjdtddtjdtddtjdtd dtjd gS NrZdenoiser_input_fieldszThe prompt embeddings)r kwargs_typerrprompt_embeds_maskzThe encoder attention masknegative_prompt_embedszThe negative prompt embeddingsnegative_prompt_embeds_maskz#The negative prompt embeddings maskrr'rrs r.rz-QwenImageTextEncoderStep.intermediate_outputst $3,,3   )3,,8   -3,,<   23,,A  '  r0ct|ts't|tstdt ||7t|ts't|tstdt |||dkDrtd|yy)N2`prompt` has to be of type `str` or `list` but is ;`negative_prompt` has to be of type `str` or `list` but is rz9`max_sequence_length` cannot be greater than 1024 but is rCrDrlrr)r1rrs r. check_inputsz%QwenImageTextEncoderStep.check_inputss&#&z&$/GQRVW]R^Q_`a a  '45Z[_`o[pZqrs s  */BT/IXYlXmno o0J *r0rrc |j|}|j}|j|j|j|j t |j|j|j|jj|jj|jj|\|_ |_|jddd|j f|_ |jddd|j f|_d|_d|_|j"r|jxsd}t |j|j||jj|jj|jj|\|_|_|jddd|j f|_|j ddd|j f|_|j%||||fS)N)r1r2r3r4r5rk)r_execution_devicerr1rrr\rNrOrr2r3r4rZrrrrequires_unconditional_embedsrrrrrr5rs r.rz!QwenImageTextEncoderStep.__call__s**51 -- +,,k.I.I;KjKjkDZ  # #  %%#-#4#4#K#K-7->->-_-_!+!2!2!G!GE A !;#A%0$=$=aAb;CbCbAb>b$c !)4)G)GKl[MlMlKlHl)m &-1 *26 /  3 3)99?ROZp''$$&'1'8'8'O'O1;1B1B1c1c%/%6%6%K%K[ WK . 0W2=1S1S4[44442K .7B6]6]4[44447K 3 UK05  r0NrrrrrrDrrrrrrrrrr staticmethodrr'rrrrrr0r.rrsJ _S__  T-%8     $z"2   Z(   d;&7  8 p pU]]_(!#;(!M(!(!r0rceZdZdZedefdZedeefdZ edee fdZ edee fdZ edeefdZedZej&d ed efd Zy ) QwenImageEditTextEncoderSteprrcy)NzxText Encoder step that processes both prompt and image together to generate text embeddings for guiding image generationrrs r.rz(QwenImageEditTextEncoderStep.descriptionsJr0c |tdttdttdtt ddidgS)NrNrerrrrr)rrr r r rs r.rz0QwenImageEditTextEncoderStep.expected_componentssC .*L M +'7 8 &!#3S"9:(5    r0c6tddtddgS)Nr2<|im_start|>system Describe the key features of the input image (color, shape, size, texture, objects, background), then explain how the user's text instruction should alter or modify the image. Generate a new image that meets the user's requirements while maintaining consistency with the original input where appropriate.<|im_end|> <|im_start|>user <|vision_start|><|image_pad|><|vision_end|>{}<|im_end|> <|im_start|>assistant rr3@rrs r.rz-QwenImageEditTextEncoderStep.expected_configs&s, -M  > K   r0ctddtdtdtdtddtjd gS) Nr1Trrrrrrz?The image prompt to encode, should be resized using resize step)rrDr'rrs r.rz#QwenImageEditTextEncoderStep.inputs0sC HtsPf g -Ji j $,,]    r0c tddtjdtddtjdtddtjdtd dtjd gSrrrs r.rz1QwenImageEditTextEncoderStep.intermediate_outputs=rr0ct|ts't|tstdt ||9t|ts(t|tstdt |yyy)Nrrr)r1rs r.rz)QwenImageEditTextEncoderStep.check_inputsZsv&#&z&$/GQRVW]R^Q_`a a  '45Z[_`o[pZqrs s65 (r0rrc |j|}|j|j|j|j}t |j |j|j|j|jj|jj|\|_ |_ d|_d|_|j ru|jxsd}t |j |j||j|jj|jj|\|_|_|j#||||fS)N)r1r]r2r3r5 )rrr1rr rhrNrerrr2r3rZrrrr rr s r.rz%QwenImageEditTextEncoderStep.__call__fs1**51  +,,k.I.IJ--D_  # #  %%++#-#4#4#K#K-7->->-_-_E A !;#A.2 *26 /  3 3)99@SOZu''$$&!//'1'8'8'O'O1;1B1B1c1c[ WK . 0W UK05  r0Nr rr0r.rrsJ JSJJ  T-%8     $z"2    Z(     d;&7  8 t tU]]_ !#; !M ! !r0rcdeZdZdZedeefdZejde de fdZ y) QwenImageEditPlusTextEncoderSteprrcNtddtddtddgS)Nr2<|im_start|>system Describe the key features of the input image (color, shape, size, texture, objects, background), then explain how the user's text instruction should alter or modify the image. Generate a new image that meets the user's requirements while maintaining consistency with the original input where appropriate.<|im_end|> <|im_start|>user {}<|im_end|> <|im_start|>assistant rri7Picture {}: <|vision_start|><|image_pad|><|vision_end|>r3rrrs r.rz1QwenImageEditPlusTextEncoderStep.expected_configss< -b  *Q  > K  r0rrc |j|}|j|j|j|j}t |j |j|j|j|jj|jj|jj|\|_ |_d|_d|_|j"r|jxsd}t |j |j||j|jj|jj|jj|\|_|_|j%||||fS)N)r1r]r2rir3r5r)rrr1rr rqrNrerrr2rir3rZrrrr rr s r.rz)QwenImageEditPlusTextEncoderStep.__call__sQ**51  +,,k.I.IJ--Dd  # #  %%++#-#4#4#K#K * 1 1 E E-7->->-_-_ E A !;#A.2 *26 /  3 3)99@SO0++((*%33+5+<+<+S+S(2(9(9(M(M5?5F5F5g5g!  XK . 0W UK05  r0N) rrrrrrrrr'rrrrrr0r.rrsPJ   $z"2    U]]_$!#;$!M$!$!r0rceZdZdZedefdZedeefdZ edee fdZ edee fdZ edZej"ded efd Zy ) &QwenImageInpaintProcessImagesInputSteprrcy)NzImage Preprocess step for inpainting task. This processes the image and mask inputs together. Images can be resized first using QwenImageEditResizeDynamicStep.rrs r.rz2QwenImageInpaintProcessImagesInputStep.descriptionsqr0c@tdttddidgS)Nimage_mask_processorrrrr)rr r rs r.rz:QwenImageInpaintProcessImagesInputStep.expected_componentss- & !#5r":;(5    r0ctddtdtdtdtdtdgS) N mask_imageTrrr]rrpadding_mask_croprrs r.rz-QwenImageInpaintProcessImagesInputStep.inputssA |d 3  ' w  x w  * +   r0cTtdtdtdtdgS)Nprocessed_imagerprocessed_mask_imagemask_overlay_kwargsz=The kwargs for the postprocess step to apply the mask overlayr)rrrs r.rz;QwenImageInpaintProcessImagesInputStep.intermediate_outputss1 . / 3 4 *[   r0c|||dzzdk7rtd|dzd|| ||dzzdk7rtd|dzd|yyNrrzHeight must be divisible by z but is zWidth must be divisible by rrrrs r.rz3QwenImageInpaintProcessImagesInputStep.check_inputs  &,!?1!D:;Ka;O:PPXY^X_`a a"E r0rrcr|j|}|j|j td|js|j}|j |j |j |j|j xs |j}|j xs |j}n(|jdj\}}|j}|jj||j|||j\|_|_|_|j%||||fS)N7resized_image and image cannot be None at the same timer3r)r]r!rrr))rrr]rrrrrdefault_height default_widthrHr% preprocessr'r)r,r.r/rrrrrr]rrs r.rz/QwenImageInpaintProcessImagesInputStep.__call__s4**51  $ $ ,1B1B1JVW W  $ $ ,%%E   "))1B1BU_UpUp  !''D:+D+DF%%A)A)AE'55a8==ME6--E  + + 6 6 ++"-"?"? 7  g #[%E{Gf UK05  r0NrrrrrrDrrrrrrrrrrr'rrrrrr0r.r"r"sJ qSqq T-%8   Z(    d;&7    bbU]]_!#;!M!!r0r"ceZdZdZedefdZedeefdZ edee fdZ edee fdZ edZej"ded efd Zy ) QwenImageProcessImagesInputSteprrcy)NzXImage Preprocess step. Images can be resized first using QwenImageEditResizeDynamicStep.rrs r.rz+QwenImageProcessImagesInputStep.descriptionsir0c@tdttddidgS)Nimage_processorrrrrrrs r.rz3QwenImageProcessImagesInputStep.expected_componentss- !!!#5r":;(5    r0cVtdtdtdtdgS)Nrr]rrr*rs r.rz&QwenImageProcessImagesInputStep.inputs&s'?+Z-@*XBVXbcjXkllr0ctdgS)Nr,r-)rrs r.rz4QwenImageProcessImagesInputStep.intermediate_outputs*s . /  r0c|||dzzdk7rtd|dzd|| ||dzzdk7rtd|dzd|yyr1r2r3s r.rz,QwenImageProcessImagesInputStep.check_inputs0r4r0rrc*|j|}|j|j td|js|j}|j |j |j |j|j xs |j}|j xs |j}n(|jdj\}}|j}|jj||||_ |j||||fS)Nr6r3rr]rr)rrr]rrrrrr7r8rHr@r9r,rr:s r.rz(QwenImageProcessImagesInputStep.__call__8s**51  $ $ ,1B1B1JVW W  $ $ ,%%E   "))1B1BU_UpUp  !''D:+D+DF%%A)A)AE'55a8==ME6--E&0&@&@&K&K'L' # UK05  r0Nr;rr0r.r=r=sJ jSjj T-%8  mZ(mm d;&7  bbU]]_!#;!M!!r0r=c~eZdZdZdZedefdZedee fdZ e jde defdZy ) 'QwenImageEditPlusProcessImagesInputStepzqwenimage-edit-plusrrcy)NzImage Preprocess step for QwenImage Edit Plus. Unlike QwenImage Edit, QwenImage Edit Plus doesn't use the same resized image for further preprocessing.rrs r.rz3QwenImageEditPlusProcessImagesInputStep.descriptionXsir0cVtdtdtdtdgS)Nrr]rrr*rs r.rz.QwenImageEditPlusProcessImagesInputStep.inputs\s';'G)RT^_fTghhr0rrcp|j|}|j|j td|j|j}|j |j |j |j|j xs |j}|j xs |j}|jj||||_ nK|jdj\}}|j}|jj||||_ |j||||fS)Nz7`vae_image` and `image` cannot be None at the same timer3rEr)rrr]rrrrrr7r8r@r9r,rHrr:s r.rz0QwenImageEditPlusProcessImagesInputStep.__call__`s7**51  ([->->-FVW W  (%%E   "))1B1BU_UpUp  !''D:+D+DF%%A)A)AE*4*D*D*O*OF%+P+K '(11!499ME6))E*4*D*D*O*OF%+P+K ' UK05  r0N)rrrrvae_image_sizerrDrrrrr'rrrrrr0r.rGrGTsv&J N iSiiiZ(iiU]]_!#;!M!!r0rGceZdZdZ d dedeffd ZedefdZedee fdZ edee fdZ edee fd Zej d ed edefd ZxZS)QwenImageVaeEncoderDynamicSteprrrc>||_||_t| y)aInitialize a VAE encoder step for converting images to latent representations. Both the input and output names are configurable so this block can be configured to process to different image inputs (e.g., "processed_image" -> "image_latents", "processed_control_image" -> "control_image_latents"). Args: input_name (str, optional): Name of the input image tensor. Defaults to "processed_image". Examples: "processed_image" or "processed_control_image" output_name (str, optional): Name of the output latent tensor. Defaults to "image_latents". Examples: "image_latents" or "control_image_latents" Examples: # Basic usage with default settings (includes image processor) QwenImageVaeEncoderDynamicStep() # Custom input/output names for control image QwenImageVaeEncoderDynamicStep( input_name="processed_control_image", output_name="control_image_latents" ) N)r_image_latents_output_namerrrs r.rz'QwenImageVaeEncoderDynamicStep.__init__s .",*5' r0rc<d|jd|jdS)Nz'Dynamic VAE Encoder step that converts z into latent representations z. )rrOrs r.rz*QwenImageVaeEncoderDynamicStep.descriptions>89O9O8PPmnroNoNnOORS Sr0c(tdtg}|S)Nr~)rrrrs r.rz2QwenImageVaeEncoderDynamicStep.expected_componentss %!7 8 r0cJt|jdtdg}|S)NTr(rs)rrrrs r.rz%QwenImageVaeEncoderDynamicStep.inputss* t-- = { #  r0cPt|jtjdgS)Nz,The latents representing the reference imagerr)rrOr'rrs r.rz3QwenImageVaeEncoderDynamicStep.intermediate_outputss) //,,J   r0rrcL|j|}|j}|jj}t ||j }t ||j|j|||j}t||j||j||||fS)N)r]r~rsr5rAr) rr r~rArrrrsnum_channels_latentsrrOr)rrrrr5rAr]rs r.rz'QwenImageVaeEncoderDynamicStep.__call__s**51 --$$ T%;%;<)!++&;;    T<>II!//JM 1A#NN%// * ? ?$1K -mnrs}n~mA  UK05  r0Nr;rr0r.rZrZsJ uSuu T-%8  Z( d;&7  bbU]]_:!#;:!M:!Vc:!:!r0rZ)NrrrN)NNrrN)NNrrrN)Nrw)rrx)Atypingrrrrrr' transformersrrr configuration_utilsr guidersr r@r rrrmodelsrrr+pipelines.qwenimage.pipeline_qwenimage_editrutilsrutils.torch_utilsrmodular_pipelinerrmodular_pipeline_utilsrrrrr get_loggerrloggerrr/rDrr5r\rhrrq Generatorr}rArrrrrrr"r=rGrMrZrr0r.rts/. ]]--lleeO.CWW6   H %%,,ell%)#A,. $%)%1 #tCy. !%1 %1 '* %1  %1 U\\ "%1V%)$(#W,.%),1 #tCy. !,1 ELL ! ,1  ,1 '* ,1 U\\ ",1d%)SW#lX,.%)41 #tCy. !41 E%,,SYY__(=syyNO P 41  41  41'*41 U\\ "41rck TLL T-5eoo-F T\_ T(( <<( (( LL ( ;; (  ((VM!%:M!`J!)GJ!Z~!4~!Bu!#8u!p6!'C6!rN!-BN!b=!&;=!@&!.M&!RQ!%:Q!hl!(=l!r0