3j >SSKrSSKJr SSKJrJr SSKJrJrJ r SSKJ r J r J r SS \ S \RS \S \S \R4 SjjrS\RS\S \\\44Sjr"SS\5r"SS\5r"SS\5r"SS\5r"SS\5rg)N)QwenImageMultiControlNetModel)ModularPipelineBlocks PipelineState) ComponentSpec InputParam OutputParam)QwenImageLayeredPachifierQwenImageModularPipelineQwenImagePachifier input_name input_tensor batch_sizenum_images_per_promptreturnc  [U[R5(d[SUS35eURSS:XaX#-nO7URSU:XaUnO![SUSUSURS35eUR USS9nU$)aRepeat tensor elements to match the final batch size. This function expands a tensor's batch dimension to match the final batch size (batch_size * num_images_per_prompt) by repeating each element along dimension 0. The input tensor must have batch size 1 or batch_size. The function will: - If batch size is 1: repeat each element (batch_size * num_images_per_prompt) times - If batch size equals batch_size: repeat each element num_images_per_prompt times Args: input_name (str): Name of the input tensor (used for error messages) input_tensor (torch.Tensor): The tensor to repeat. Must have batch size 1 or batch_size. batch_size (int): The base batch size (number of prompts) num_images_per_prompt (int, optional): Number of images to generate per prompt. Defaults to 1. Returns: torch.Tensor: The repeated tensor with final batch size (batch_size * num_images_per_prompt) Raises: ValueError: If input_tensor is not a torch.Tensor or has invalid batch size Examples: tensor = torch.tensor([[1, 2, 3]]) # shape: [1, 3] repeated = repeat_tensor_to_batch_size("image", tensor, batch_size=2, num_images_per_prompt=2) repeated # tensor([[1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3]]) - shape: [4, 3] tensor = torch.tensor([[1, 2, 3], [4, 5, 6]]) # shape: [2, 3] repeated = repeat_tensor_to_batch_size("image", tensor, batch_size=2, num_images_per_prompt=2) repeated # tensor([[1, 2, 3], [1, 2, 3], [4, 5, 6], [4, 5, 6]]) - shape: [4, 3] `z` must be a tensorrr z!` must have have batch size 1 or z , but got dim) isinstancetorchTensor ValueErrorshaperepeat_interleave)rrrr repeat_bys f/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/modular_pipelines/qwenimage/inputs.pyrepeat_tensor_to_batch_sizer sJ lELL 1 11ZL(:;<<!!6   A * ,)  |   r!c [S[SS9[S[RSS9[R"SSS9[R"S SS9[R"S SS9[R"S SS9/$) Nrz'The batch size of the prompt embeddingsname type_hintr7dtypez&The data type of the prompt embeddingsr:zbatch-expanded)noter;r<r=)r intrrFr>r?s rintermediate_outputs,QwenImageTextInputsStep.intermediate_outputsss \SFo p W Iq r  7G H  !5EU V   r!cXUbUc [S5eUcUb [S5eURSURS:wa [S5eUb+URSURS:wa [S5eUb,URSURS:wa [S5egg)NzS`negative_prompt_embeds_mask` is required when `negative_prompt_embeds` is not NonezJcannot pass `negative_prompt_embeds_mask` without `negative_prompt_embeds`rzE`prompt_embeds_mask` must have the same batch size as `prompt_embeds`zI`negative_prompt_embeds` must have the same batch size as `prompt_embeds`zN`negative_prompt_embeds_mask` must have the same batch size as `prompt_embeds`)rrr:r;r<r=s r check_inputs$QwenImageTextInputsStep.check_inputss " -2M2Urs s ! ).I.Uij j  # #A &-*=*=a*@ @de e # /4J4P4PQR4SWdWjWjklWm4mhi i ( 38S8Y8YZ[8\`m`s`stu`v8vmn n9w 3r! componentsstatecURU5nURURURURUR S9 URR SUlURRUlURR upEnURRSURS5UlURRURUR-US5UlURRSURS5UlURRURUR-U5UlURbURR upEnURRSURS5UlURRURUR-US5UlUR RSURS5UlUR RURUR-U5UlURX#5 X4$)NrLrr ) get_block_staterMr:r;r<r=rrrFrepeatrviewset_block_state)r4rOrP block_state_seq_lens r__call__ QwenImageTextInputsStep.__call__s***51  %33*==#.#E#E(3(O(O  "-!:!:!@!@!C '55;; #1177 A$/$=$=$D$DQ HiHikl$m !$/$=$=$B$B  " "[%F%F FQS% !*5)G)G)N)NqR]RsRsuv)w &)4)G)G)L)L  " "[%F%F F* &  - - 9'>>DDMA1<1S1S1Z1Z;44a2K .2=1S1S1X1X&&)J)JJGUW2K .7B6]6]6d6d;44a7K 37B6]6]6b6b&&)J)JJG7K 3 U0  r!r3N)__name__ __module__ __qualname____firstlineno____doc__ model_namepropertystrr7listr r@r rI staticmethodrMr rrZ__static_attributes__r3r!rr/r/ps"HJ  3S 3 3 Z(   d;&7  oo.+!#;+!M+!Vc+!r!r/c^\rSrSrSrSrSS\\S-S\\S-4U4Sjjjr\ S\ 4S j5r \ S\\ 4S j5r \ S\\4S j5r\ S\\4S j5rS \S\S\4SjrSrU=r$)QwenImageAdditionalInputsStepia1 Input processing step that: 1. For image latent inputs: Updates height/width if None, patchifies, and expands batch size 2. For additional batch inputs: Expands batch dimensions to match final batch size Configured inputs: - Image latent inputs: ['image_latents'] This block should be placed after the encoder steps and the text input step. Components: pachifier (`QwenImagePachifier`) Inputs: num_images_per_prompt (`int`, *optional*, defaults to 1): The number of images to generate per prompt. batch_size (`int`, *optional*, defaults to 1): Number of prompts, the final batch size of model inputs should be batch_size * num_images_per_prompt. Can be generated in input step. height (`int`, *optional*): The height in pixels of the generated image. width (`int`, *optional*): The width in pixels of the generated image. image_latents (`Tensor`): image latents used to guide the image generation. Can be generated from vae_encoder step. Outputs: image_height (`int`): The image height calculated from the image latents dimension image_width (`int`): The image width calculated from the image latents dimension height (`int`): if not provided, updated to image height width (`int`): if not provided, updated to image width image_latents (`Tensor`): image latents used to guide the image generation. Can be generated from vae_encoder step. (patchified and batch-expanded) r1Nimage_latent_inputsadditional_batch_inputsc>Uc[R"S5/nUc/n[U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e [U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e XlX l[TU]%5 gN image_latentsz,image_latent_inputs must be a list, but got z:image_latent_inputs must be a list of InputParam, but got z0additional_batch_inputs must be a list, but got z>additional_batch_inputs must be a list of InputParam, but got r r>rrdrtype_image_latent_inputs_additional_batch_inputssuper__init__r4rirj input_param __class__s rrs&QwenImageAdditionalInputsStep.__init__,s  &#-#6#6#G"H  " *&( #-t44KDQdLeKfgh h2 !+z::$'abfgrbsat%uvv 31488OPTUlPmOnop p6 !+z::$XY]^iYjXkl 7 %8!(?% r!rclSnSnUR(dUR(a|SnUR(a,USURVs/sHo3RPM sn3- nUR(a,USURVs/sHo3RPM sn3- nSnX-U-$s snfs snf)NzInput processing step that: 1. For image latent inputs: Updates height/width if None, patchifies, and expands batch size 2. For additional batch inputs: Expands batch dimensions to match final batch size Configured inputs: - Image latent inputs:  - Additional batch inputs: N This block should be placed after the encoder steps and the text input step.rprqrDr4r5 inputs_infopr6s rr7)QwenImageAdditionalInputsStep.descriptionK c    $ $(E(E2K((!iAp B,B1c"[S[SS9/$N pachifier from_config)default_creation_methodrrr?s rexpected_components1QwenImageAdditionalInputsStep.expected_components_ +'9S` a  r!c[R"S5[R"S5[R"S5[R"S5/nXRUR-- nU$Nrrr+r,r r>rprqr4r@s rr@$QwenImageAdditionalInputsStep.inputsesd    7 8    -    )    (   ++d.K.KKK r!c $[S[SS9[S[SS9/n[UR5S:a<UR [S[SS95 UR [S [S S95 URH>nUR [UR UR URS -S95 M@ URH>nUR [UR UR URS -S95 M@ U$) N image_height20&1&G&G&11 #  K:M N75< 88K$))J";;L#6%)&1&G&G&11 L K\ :9 U0  r!rqrpNNr\r]r^r_r`rardr rsrbrcr7rrr@r rIr rrZrf __classcell__rvs@rrhrhs&PJ8<;?!*-4"&j!1D!8>ASAA& T-%8   Z(  +d;&7++Z2!#;2!M2!Vc2!2!r!rhc^\rSrSrSrSrSS\\S-S\\S-4U4Sjjjr\ S\ 4S j5r \ S\\ 4S j5r \ S\\4S j5r\ S\\4S j5rS \S\S\4SjrSrU=r$)%QwenImageEditPlusAdditionalInputsStepia Input processing step for Edit Plus that: 1. For image latent inputs (list): Collects heights/widths, patchifies each, concatenates, expands batch 2. For additional batch inputs: Expands batch dimensions to match final batch size Height/width defaults to last image in the list. Configured inputs: - Image latent inputs: ['image_latents'] This block should be placed after the encoder steps and the text input step. Components: pachifier (`QwenImagePachifier`) Inputs: num_images_per_prompt (`int`, *optional*, defaults to 1): The number of images to generate per prompt. batch_size (`int`, *optional*, defaults to 1): Number of prompts, the final batch size of model inputs should be batch_size * num_images_per_prompt. Can be generated in input step. height (`int`, *optional*): The height in pixels of the generated image. width (`int`, *optional*): The width in pixels of the generated image. image_latents (`Tensor`): image latents used to guide the image generation. Can be generated from vae_encoder step. Outputs: image_height (`list`): The image heights calculated from the image latents dimension image_width (`list`): The image widths calculated from the image latents dimension height (`int`): if not provided, updated to image height width (`int`): if not provided, updated to image width image_latents (`Tensor`): image latents used to guide the image generation. Can be generated from vae_encoder step. (patchified, concatenated, and batch-expanded) zqwenimage-edit-plusNrirjc>Uc[R"S5/nUc/n[U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e [U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e XlX l[TU]%5 grlrnrts rrs.QwenImageEditPlusAdditionalInputsStep.__init__  &#-#6#6#G"H  " *&( #-t44KDQdLeKfgh h2 !+z::$'abfgrbsat%uvv 31488OPTUlPmOnop p6 !+z::$XY]^iYjXkl 7 %8!(?% r!rclSnSnUR(dUR(a|SnUR(a,USURVs/sHo3RPM sn3- nUR(a,USURVs/sHo3RPM sn3- nSnX-U-$s snfs snf)NaInput processing step for Edit Plus that: 1. For image latent inputs (list): Collects heights/widths, patchifies each, concatenates, expands batch 2. For additional batch inputs: Expands batch dimensions to match final batch size Height/width defaults to last image in the list.ryrzr{r|r}r~rs rr71QwenImageEditPlusAdditionalInputsStep.description s A   $ $(E(E2K((!iAprc"[S[SS9/$rrr?s rr9QwenImageEditPlusAdditionalInputsStep.expected_components5rr!c[R"S5[R"S5[R"S5[R"S5/nXRUR-- nU$rrrs rr@,QwenImageEditPlusAdditionalInputsStep.inputs;sd    7 8    -    )    (   ++d.K.KKK r!c @[S[[SS9[S[[SS9/n[UR5S:a<UR [S[SS95 UR [S [S S95 URH>nUR [UR URURS -S95 M@ URH>nUR [UR URURS -S95 M@ U$) Nrz=The image heights calculated from the image latents dimensionrCrzUc[R"S5/nUc/n[U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e [U[5(d[ S[ U535eUH/n[U[5(aM[ S[ U535e XlX l[TU]%5 grlrnrts rrs-QwenImageLayeredAdditionalInputsStep.__init__rr!rclSnSnUR(dUR(a|SnUR(a,USURVs/sHo3RPM sn3- nUR(a,USURVs/sHo3RPM sn3- nSnX-U-$s snfs snf)NzInput processing step for Layered that: 1. For image latent inputs: Updates height/width if None, patchifies with layered pachifier, and expands batch size 2. For additional batch inputs: Expands batch dimensions to match final batch sizeryrzr{r|r}r~rs rr70QwenImageLayeredAdditionalInputsStep.descriptionrrc"[S[SS9/$r)rr r?s rr8QwenImageLayeredAdditionalInputsStep.expected_componentss +'@Zg h  r!c[R"S5[R"S5/nXRUR-- nU$)Nrrrrs rr@+QwenImageLayeredAdditionalInputsStep.inputs sH    7 8    -  ++d.K.KKK r!c $[S[SS9[S[SS9/n[UR5S:a<UR [S[SS95 UR [S [S S95 URH>nUR [UR UR URS -S95 M@ URH>nUR [UR UR URS -S95 M@ U$) NrrrCrrrr+rr,rz7 (patchified with layered pachifier and batch-expanded)rrrs rrI9QwenImageLayeredAdditionalInputsStep.intermediate_outputs,s  #Z  "Y    t(( )A - NNSFpq  NNCEno   44K NN$)))33 + 7 7:s s 5 88K NN$)))33 + 7 7:M M 9r!rOrPcURU5nURHnURn[X55nUcMURSUR -nURSUR -nXslXl[US5(dXsl [US5(dXl URRU5n[UUURURS9n[!X5U5 M UR"HKnURn [X95n U cM[U U URURS9n [!X9U 5 MM UR%X#5 X4$)Nrr%rrr)rSrprDrrr#r+r,rrrrrr rrrrqrVrs rrZ-QwenImageLayeredAdditionalInputsStep.__call__Ysr**51  44K&1&6&6 #")+"O "*)..q1J4O4OOF'--a0:3N3NNE!'  % ;77+1(; 66*/'#-"6"6"C"CDW"X #>20&1&G&G&11 #  K:M N;5@ 88K$))J";;L#6%)&1&G&G&11 L K\ :9 U0  r!rrrrs@rrrs#J%J8<;?!*-4"&j!1D!8<ASAA& T-%8   Z(  *d;&7**X4!#;4!M4!Vc4!4!r!rc\rSrSrSrSr\S\4Sj5r\S\ \ 4Sj5r \S\ \ 4Sj5r \R"5S\S \S\4S j5rS rg ) QwenImageControlNetInputsStepia! prepare the `control_image_latents` for controlnet. Insert after all the other inputs steps. Inputs: control_image_latents (`Tensor`): The control image latents to use for the denoising process. Can be generated in controlnet vae encoder step. batch_size (`int`, *optional*, defaults to 1): Number of prompts, the final batch size of model inputs should be batch_size * num_images_per_prompt. Can be generated in input step. num_images_per_prompt (`int`, *optional*, defaults to 1): The number of images to generate per prompt. height (`int`, *optional*): The height in pixels of the generated image. width (`int`, *optional*): The width in pixels of the generated image. Outputs: control_image_latents (`Tensor`): The control image latents (patchified and batch-expanded). height (`int`): if not provided, updated to control image height width (`int`): if not provided, updated to control image width r1rcg)Nz\prepare the `control_image_latents` for controlnet. Insert after all the other inputs steps.r3r?s rr7)QwenImageControlNetInputsStep.descriptionsmr!c[SS[RSS9[R"S5[R"S5[R"S5[R"S5/$) Ncontrol_image_latentsTzlThe control image latents to use for the denoising process. Can be generated in controlnet vae encoder step.)rDrequiredrEr7rrr+r,)r rrr>r?s rr@$QwenImageControlNetInputsStep.inputssd ,,,K      -    7 8    )    (  r!cn[S[RSS9[S[SS9[S[SS9/$)Nrz:The control image latents (patchified and batch-expanded).rCr+z0if not provided, updated to control image heightr,z/if not provided, updated to control image width)r rrrHr?s rrI2QwenImageControlNetInputsStep.intermediate_outputss? ,,,X  XBt u WAr s  r!rOrPcURU5n[UR[5(a/n[ UR 5HupV[ XaR5upxUR=(d UUlUR=(d UUl URRU5n[SUS3UURURS9nURU5 M XClO[ UR UR5upxUR=(d UUlUR=(d UUl URRUR 5Ul[SUR URURS9UlUR UlUR!X#5 X4$)Nzcontrol_image_latents[rrr)rSr controlnetrrrr-r#r+r,rrr rrrrV) r4rOrPrWrrcontrol_image_latents_r+r,s rrZ&QwenImageControlNetInputsStep.__call__s**51 j++-J K K$& !-6{7X7X-Y) @AWYtYt u %0%7%7%A6 "$/$5$5$> !*4)=)=)J)JKa)b&*E!7s!>&1&G&G&11 1K -1<0Q0QK - U0  r!r3N)r\r]r^r_r`rarbrcr7rdr r@r rIrno_gradr rrZrfr3r!rrrs4J nSnn  Z(      d;&7     ]]_2!#;2!M2!Vc2!2!r!r)r )rmodelsrmodular_pipelinerrmodular_pipeline_utilsrr r r r rrcrrHr tupler-r/rhrrrr3r!rrs  3CKKee"# 55,,55 5  \\ 5pellcV[\_ad\dVe@M!3M!bQ!$9Q!jc!,Ac!TM!+@M!bo!$9o!r!