3j[.`SSKrSSKrSSKJs Jr SSKJrJr SSK J r SSSSS .r "S S \RR5r "S S \RR5r"SS\RR5r"SS\RR5r"SS\ \5rg)N) ConfigMixinregister_to_config) ModelMixin)r)r)r)rr )g?g?@g@c|^\rSrSrS S\S\S-S\4U4SjjjrS\RS\R4S jrS r U=r $) ResBlock!Nchannels mid_channelsdimsc>[TU]5 UcUnUS:Xa[RRO[RR nU"XSSS9Ul[RRSU5UlU"X!SSS9Ul [RRSU5Ul [RR5Ul g)Nrrr  kernel_sizepadding ) super__init__torchnnConv2dConv3dconv1 GroupNormnorm1conv2norm2SiLU activation)selfrrrConv __class__s c/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/pipelines/ltx2/latent_upsampler.pyrResBlock.__init__"s   #L"&!)uxx(aK XX''L9 ,aK XX''H5 ((--/ hidden_statesreturncUnURU5nURU5nURU5nURU5nUR U5nURX-5nU$N)rrr"rr )r#r)residuals r&forwardResBlock.forward/sd  =1  =1  6  =1  =1  (@A r()r"rrrr )Nr) __name__ __module__ __qualname____firstlineno__intrrTensorr.__static_attributes__ __classcell__r%s@r&r r !sF * *C$J *S * *U\\ellr(r c2^\rSrSrSU4SjjrSrSrU=r$)PixelShuffleND;c\>[TU]5 XlX lUS;a [ S5eg)N)r rrzdims must be 1, 2, or 3)rrrupscale_factors ValueError)r#rr=r%s r&rPixelShuffleND.__init__<s2  . y 67 7 !r(c rURS:XaiURSS/URSSQ75RSSSSSSSS 5R SS5R S S5R SS5$URS:XaWURSS/URSSQ75RSSS SSS5R S S5R SS5$URS:XaGURSS/URSSQ75RSSSSS S5R SS5$g) Nrr rrr)r unflattenr=permuteflattenr#xs r&r.PixelShuffleND.forwardEs= 99> A>T%9%9"1%=>?Aq!Q1a0AAA  YY!^ A>T%9%9"1%=>?GG1aQRTUWXYaabcefgoopqstu YY!^;;q2"A(<([TU]5 US;a[SU35eUS:d US-S:wa[SU35eXlX lX0l[ R"[U5Vs/sHn[R"US- U5PM sn5nUSS2S4USSS24-nXUR5- R5nURSUSSSS2SS245 gs snf)N)rrz$`dims` must be either 2 or 3 but is rrr z0`kernel_size` must be an odd number >= 3 but is kernel)rrr>rrNrrtensorrangemathcombsumfloatregister_buffer)r#rrNrkk2dr%s r&rBlurDownsample.__init___s  v CD6JK K ?kAo2OP[}]^ ^  & LL{ASTASA$))K!OQ7AST U4j1T1W:%WWY%%' Xs4q!+;'<=Us0$C,rIc URS:XaU$URS:XarURSnURR USUR UR 5n[ R"XSURUR S-US9nU$URupBn nURSS5RSS5nURR USUR UR 5n[ R"XSURUR S-US9nURSSupxURSXE45RUSXWU5nU$)Nr r)weightbiasrNrgroupsrrA) rNrshaperPexpandrFconv2d transposerGrEreshape) r#rIcr\bf_h2w2s r&r.BlurDownsample.forwardts7 ;;! H 99> A[[''1d.>.>@P@PQFT[[RVRbRbfgRgpqrAGGMA!Q Aq!))!Q/A[[''1d.>.>@P@PQFT[[RVRbRbfgRgpqrAWWRS\FB Av&..q"aR@Ar()rrrN)rB) r0r1r2r3__doc__r4rrr5r.r6r7r8s@r&rLrLYsN >S>#>C>>>*%,,r(rLcv^\rSrSrSrS S\S\4U4SjjjrS\RS\R4Sjr S r U=r $) SpatialRationalResamplerak Scales by the spatial size of the input by a rational number `scale`. For example, `scale = 0.75` will downsample by a factor of 3 / 4, while `scale = 1.5` will upsample by a factor of 3 / 2. This works by first upsampling the input by the (integer) numerator of `scale`, and then performing a blur + stride anti-aliased downsample by the (integer) denominator. rscalec >[TU]5 [U5Ul[R US5nUc,[ SUS[[R5535eUuUl Ul [RRXRS-U-SSS9Ul[SURUR4S9Ul[#SURS9Ulg) NzThe supplied `scale` z( is not supported; supported scales are rrr r)r=)rrN)rrrVrq RATIONAL_RESAMPLER_SCALE_MAPPINGgetr>listkeysnumdenrrrconvr: pixel_shufflerL blur_down)r#rrq num_denomr%s r&r!SpatialRationalResampler.__init__s 5\ 488E  'w.VW[\|]B]B]DXEWFG '$(HHOOL88Q;,2N\]ghOi +A$((?ST'Qtxx@r(rIr*clURU5nURU5nURU5nU$r,)ryrzr{rHs r&r. SpatialRationalResampler.forwards4 IIaL   q ! NN1 r()r{ryrxrwrzrq)r ) r0r1r2r3rmr4rVrrr5r.r6r7r8s@r&rorosD AS A A A%,,r(roc^\rSrSrSr\SS\S\S\S\S\S\S \S \4U4S jjj5r S \ RS \ R4Sjr Sr U=r$)LTX2LatentUpsamplerModela Model to spatially upsample VAE latents. Args: in_channels (`int`, defaults to `128`): Number of channels in the input latent mid_channels (`int`, defaults to `512`): Number of channels in the middle layers num_blocks_per_stage (`int`, defaults to `4`): Number of ResBlocks to use in each stage (pre/post upsampling) dims (`int`, defaults to `3`): Number of dimensions for convolutions (2 or 3) spatial_upsample (`bool`, defaults to `True`): Whether to spatially upsample the latent temporal_upsample (`bool`, defaults to `False`): Whether to temporally upsample the latent in_channelsrnum_blocks_per_stagerspatial_upsampletemporal_upsamplerational_spatial_scaleuse_rational_resamplerc $>[T U]5 XlX lX0lX@lXPlX`lUS:Xa[RRO[RRn U "XSSS9Ul [RRSU5Ul[RR5Ul[RR#[%U5V s/sH n ['X$S9PM sn 5UlU(aWU(aP[RR+[RRUSU-SSS9[-S55UlOU(afU(a[1X'S9UlO[RR+[RRUS U-SSS9[-S55UlObU(aP[RR+[RRUSU-SSS9[-S55UlO [3S 5e[RR#[%U5V s/sH n ['X$S9PM sn 5UlU "X!SSS9Ulgs sn fs sn f) Nrrr rr)r)rrqrz9Either spatial_upsample or temporal_upsample must be True)rrrrrrrrrrrr initial_convr initial_normr!initial_activation ModuleListrRr res_blocks Sequentialr: upsamplerror>post_upsample_res_blocks final_conv) r#rrrrrrrrConvNdrir%s r&r!LTX2LatentUpsamplerModel.__init__s &($8! 0!2$(AI588??";!UVW!HH..r<@"'((--/((--Z_`tZu.vZuUVx /PZu.vw  1"XX00 a,.>AWXYq!DN%!9|!r!&!4!4HHOOL!l2BPQ[\O]"1%""XX00 a,.>AWXYq!DN XY Y(-(;(;8=>R8S T8S1Xl .8S T) %!STU7/w0 Us )JJ r)r*cURup#pEnURS:XaURSSSSS5RSS5nUR U5nUR U5nUR U5nURH nU"U5nM URU5nURH nU"U5nM URU5nURSUS45RSSSSS5nU$UR U5nUR U5nUR U5nURH nU"U5nM UR(a&URU5nUSS2SS2SS2SS2SS24nO]URSSSSS5RSS5nURU5nURSUS45RSSSSS5nURH nU"U5nM URU5nU$)Nrrr rrrA) r`rrFrGrrrrrrrrEr)r#r) batch_size num_channels num_framesheightwidthblocks r&r. LTX2LatentUpsamplerModel.forwards >K>Q>Q; *e 99>)11!Q1a@HHANM --mrs B/     $ uxx4kUXX__k<-UXX__-`uxxBsz;sr(