+h%SSKrSSKJrJrJr SSKrSSKrSSKJ r J r SSK J r SSK Jr SSKJrJr SS KJrJr SS KJrJrJrJr S S KJr \"5(aSSKr\R8"\5rS rSr Sr!"SS\5r"g)N)ListOptionalUnion) AutoTokenizerUMT5EncoderModel) FrozenDict)ClassifierFreeGuidance)is_ftfy_availablelogging)ModularPipelineBlocks PipelineState) ComponentSpec ConfigSpec InputParam OutputParam)WanModularPipelinec[R"U5n[R"[R"U55nUR 5$N)ftfyfix_texthtmlunescapestriptexts b/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/modular_pipelines/wan/encoders.py basic_cleanr %s3 == D ==t, -D ::<cV[R"SSU5nUR5nU$)Nz\s+ )resubrrs rwhitespace_cleanr&+s$ 66&#t $D ::>a>(--/"//0A0A&0I477SY?[mm %((u(D +.}+GH+G412A+G H ^k l^kYZUYY;;':VVAY'Fq RS T^k lrs )3I lsE<;F#AFNnum_videos_per_promptprepare_unconditional_embedsrCrHrLc U=(d URn[U[5(aU/OUnUb [U5OURSn Uc[ R XX5nU(aUcU=(d Sn[U[5(aX/-OUnUb;[U5[U5La$[S[U5S[U5S35eU [U5:wa![SUS[U5SUSU S 3 5e[ R XX5nURupn URS US 5nURX-U S 5nU(a(URS US 5nURX-U S 5nXg4$) a9 Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*): prompt to be encoded device: (`torch.device`): torch device num_videos_per_prompt (`int`): number of videos that should be generated per prompt prepare_unconditional_embeds (`bool`): whether to use prepare unconditional embeddings or not negative_prompt (`str` or `List[str]`, *optional*): The prompt or prompts not to guide the image generation. If not defined, one has to pass `negative_prompt_embeds` instead. Ignored when not using guidance (i.e., ignored if `guidance_scale` is less than `1`). prompt_embeds (`torch.Tensor`, *optional*): Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not provided, text embeddings will be generated from `prompt` input argument. negative_prompt_embeds (`torch.Tensor`, *optional*): Pre-generated negative text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not provided, negative_prompt_embeds will be generated from `negative_prompt` input argument. max_sequence_length (`int`, defaults to `512`): The maximum number of text tokens to be used for the generation process. rz?`negative_prompt` should be the same type to `prompt`, but got z != .z`negative_prompt`: z has batch size z, but `prompt`: zT. Please make sure that passed `negative_prompt` matches the batch size of `prompt`.r) _execution_devicerRrSlenshaper*ryrV TypeErrorrUrepeatview) rrrBr[r{r|rCrHrLrZ batch_sizebs_embedseq_len_s r encode_prompt WanTextEncoderStep.encode_promptsL7:77'44&&$*$6S[Mrsl (( 8--/CWW0   H %   |!.|!r!