+h%.SSKJrJrJr SSKrSSKJr SSKJr SSK J r SSK J r SSK Jr S S KJr S S KJr S S KJr \ "\5rSS\R.S\\R0S\4Sjjr"SS\5rg))ListOptionalUnionN)PipelineImageInput)AutoencoderKLLTXVideo) get_logger) randn_tensor)VideoProcessor)DiffusionPipeline)LTXLatentUpsamplerModel)LTXPipelineOutputencoder_output generator sample_modec[US5(a!US:XaURRU5$[US5(a US:XaURR5$[US5(a UR$[ S5e)N latent_distsampleargmaxlatentsz3Could not access latents of provided encoder_output)hasattrrrmoderAttributeError)rrrs n/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/pipelines/ltx/pipeline_ltx_latent_upsample.pyretrieve_latentsr!s}~}--+2I))00;;  / /K84K))..00  + +%%%RSScf^\rSrSrSrS\S\SS4U4SjjrS%S\\ RS \ S \\ RS \\ RS \\ RS \\ RS\ R4SjjrS&S \ RS\ RS\4Sjjr\S&S \ RS\ RS\ RS\S\ R4 Sjj5r\S&S \ RS\ RS\ RS\S\ R4 Sjj5rSrSrSrSrSr\ R4"5S'S\\\S\ S\ S \\ RS\\\\4S\\\\\4S \S \\\ R\\ R4S!\\S"\4S#jj5r S$r!U=r"$)(LTXLatentUpsamplePipeline.vaelatent_upsamplerreturnNc>[TU]5 URXS9 [USS5bURR OSUl[USS5bURROSUl[UR S9Ul g)N)r#r$r# )vae_scale_factor) super__init__register_modulesgetattrr#spatial_compression_ratiovae_spatial_compression_ratiotemporal_compression_ratiovae_temporal_compression_ratior video_processor)selfr#r$ __class__s rr+"LTXLatentUpsamplePipeline.__init__1s  #I3:$t2L2XDHH . .^` *4;43M3YDHH / /_` + .t?a?abrvideo batch_sizedtypedevicerrc UbURXCS9$URX@RRS9n[U[5(a|[ U5U:wa[ S[ U5SUS35e[U5Vs/sH;n[URRXRS55XW5PM= nnODUV s/sH7n [URRU RS55U5PM9 nn [R"USS9RU5nURXRRURR5nU$s snfs sn f)Nr9r8z/You have passed a list of generators of length z+, but requested an effective batch size of z@. Make sure the batch size matches the length of the generators.rdim)tor#r8 isinstancelistlen ValueErrorrangerencode unsqueezetorchcat_normalize_latents latents_mean latents_std) r3r6r7r8r9rri init_latentsvids rprepare_latents)LTXLatentUpsamplePipeline.prepare_latentsBsV  ::V:9 9hhnn= i & &9~+ Ec)nEUV *|+km affp`q`q[\ 1C1CA1F!GV`q Lgllfk_b,TXX__S]]1=M-NPYZfkLlyy1588? ..|XX=R=RTXT\T\ThThi ms AE+ >E0reference_latentsfactorcpUR5n[URS55Hpn[URS55HOn[R"X%U4SS9upx[R"XEU4SS9upXEU4U - U - U-U-XEU4'MQ Mr [R "XU5nU$)a Applies Adaptive Instance Normalization (AdaIN) to a latent tensor based on statistics from a reference latent tensor. Args: latent (`torch.Tensor`): Input latents to normalize reference_latents (`torch.Tensor`): The reference latents providing style statistics. factor (`float`): Blending factor between original and transformed latent. Range: -10.0 to 10.0, Default: 1.0 Returns: torch.Tensor: The transformed latent tensor rrNr<)clonerCsizerFstd_meanlerp) r3rrPrQresultrKcr_sdr_meani_sdi_means radain_filter_latent-LTXLatentUpsamplePipeline.adain_filter_latent`s w||A'A7<<?+$~~.?1.E4P $~~fTlE !'1!6$ >$FO!t ,(GV4 rrIrJscaling_factorcURSSSSS5RURUR5nURSSSSS5RURUR5nX- U-U- nU$Nrviewr>r9r8rrIrJr_s rrH,LTXLatentUpsamplePipeline._normalize_latents|su $((B1a8;;GNNGMMZ !&&q"aA699'..'--X )^;kIrcURSSSSS5RURUR5nURSSSSS5RURUR5nX-U- U-nU$rarcres r_denormalize_latents.LTXLatentUpsamplePipeline._denormalize_latentssu $((B1a8;;GNNGMMZ !&&q"aA699'..'--X '.8<Grc8URR5 g)z Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in several steps. This is useful to save some memory and allow larger batch sizes. N)r#enable_slicingr3s renable_vae_slicing,LTXLatentUpsamplePipeline.enable_vae_slicing !rc8URR5 g)z Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to computing decoding in one step. N)r#disable_slicingrls rdisable_vae_slicing-LTXLatentUpsamplePipeline.disable_vae_slicings   "rc8URR5 g)z Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow processing larger images. N)r# enable_tilingrls renable_vae_tiling+LTXLatentUpsamplePipeline.enable_vae_tilings  rc8URR5 g)z Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to computing decoding in one step. N)r#disable_tilingrls rdisable_vae_tiling,LTXLatentUpsamplePipeline.disable_vae_tilingrorcX R-S:wdX0R-S:wa[SUSUS35eUbUb [S5eUcUc [S5egg)Nrz8`height` and `width` have to be divisible by 32 but are z and .z1Only one of `video` or `latents` can be provided.z/One of `video` or `latents` has to be provided.)r/rB)r3r6heightwidthrs r check_inputs<XLatentUpsamplePipeline.check_inputssx 66 6! ;uGiGi?imn?nWX^W__dejdkklmn n  !4PQ Q =W_NO O-=rr~rdecode_timestepdecode_noise_scale adain_factor output_type return_dictc URUUUUS9 UbSn OURSn URn Ub[U5n XR-S:waUXR-UR-S-n USU n[ R SURS[U5SU S35 URRXUS9nURU [RS 9nURUU [RU UUS 9nURX@RRURR URR"R$5nURUR&R(5nUR'U5nUS :aUR+XU5nOUnU S :Xa]UR-X@RRURR URR"R$5nUnGOURR"R.(dSnO[1URXUR(S 9n[3U[45(dU/U -nUcUnO[45(dU/U -n[R6"X\UR(S 9n[R6"XlUR(S 9SS2SSSS4nSU- U-UU--nURR9XOSS9SnURR;XS9nUR=5 U (dU4$[?US9$)N)r6r~rrrrz-Video length expected to be of the form `k * z + 1` but is z. Truncating to z frames.)r~rr;)r6r7r8r9rrlatent)rr9r8F)r)r)frames) rshape_execution_devicerAr1loggerwarningr2preprocess_videor>rFfloat32rNrhr#rIrJconfigr_r$r8r]rHtimestep_conditioningr r?r@tensordecodepostprocess_videomaybe_free_model_hooksr)r3r6r~rrrrrrrrr7r9 num_frameslatents_upsampledtimestepnoises r__call__"LTXLatentUpsamplePipeline.__call__sA     J q)J''  UJ???1D"E"EEHkHkknookz*CDDgDgChhuvyzwAvBBRS]R^^fg((99%V[9\EHHF%--H@E&&!-- ' ++ XX**DHH,@,@$((//B`B` **T22889 11': # ../@<XG'G ( "--..0D0DdhhooFdFdGE88??88$W]]i^e^k^kl!/488'6&7*&DO%-)8&#$6==*<)= )J& <<gmm\%*\\2D[b[h[h%itT4-&"11W?\/\/E%//43H"HIJ\/c]\/\/\/rr )Nr)typingrrrrFimage_processorrmodelsrutilsr utils.torch_utilsr r2r pipeline_utilsr modeling_latent_upsamplerrpipeline_outputrrrrrrrr rrrsx)( 1+--.>. H  ck TLL T-5eoo-F T\_ Tg/ 1g/r