3j3SSKrSSKrSSKrSSKrSSKrSSKJr SSKJr SSK J r SSK J r J r SSKJrJrJr SSKJr SS KJr SS KJrJr SS KJr S r\"S S9"SS 55r\"SS9"SS55r\"SS9"SS55r\"SS9"SS55r\"SS9"SS\ R@55r!g)N) LTXAVModel) NestedTensor)io)SamplerCustomAdvanced SplitSigmas)EmptyLTXVLatentVideo LTXVAddGuideLTXVCropGuides) blur_internal)LTXVAdainLatent)LTXVAddLatentGuideLTXVSelectLatents) comfy_nodecF[US5(dSUR;a [S5eURSnUSS[R"US5//nURSnUSS[R"US5//nX$4UlUR $)N raw_condsnegativez?Guider does not have negative conds, cannot use it as a guider.positiver cross_attn)hasattroriginal_conds ValueErrorcopydeepcopyr)guiderraw_posrraw_negrs K/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/easy_samplers.py_get_raw_conds_from_guiderrs 6; ' ' V22 2Q '' 3QZ -t}}WQZ/HIJ'' 3QZ -t}}WQZ/HIJ$/   LTXVBaseSampler)namecV\rSrSr\S5rSrSrSrSr S Sjr S r g) r!"c $SSS04SSS04SSS[RS S .4SS S[RS S .4SS S [RSS .4SSS04SSS04SSS04SSS04S. SSS04SSS04SSSS SS .4S!S"/S"S#S$.4SS%SS&S'S .4SSSS(S)S .4S*.S+.$),NMODELtooltipThe diffusion model to use.VAEThe VAE to use.INTi@ )defaultminmaxstepiar GUIDER/The guider to use, must be a STGGuiderAdvanced.SAMPLERThe sampler to use.SIGMASThe sigmas to use.NOISE"The noise to use for the sampling.) modelvaewidthheight num_framesrsamplersigmasnoiseIMAGE0The images to use for conditioning the sampling.STRINGz?The indices of the images to use for conditioning the sampling.FLOAT?r/The strength of the conditioning on the images.r.r/r0r'centerdisabledz$The crop mode to use for the images.)r.r'#dz2The CRF value to use for preprocessing the images. z3The blur value to use for preprocessing the images.)optional_cond_imagesoptional_cond_indicesstrengthcropcrfblurrequiredoptionalnodesMAX_RESOLUTIONss r INPUT_TYPESLTXVBaseSampler.INPUT_TYPES'sh"I/L#MN +<=>#&!$33 " #&!$33 "  "1U5I5ISTU  QR& 3H'IJ#i1E%FG!I/S#TU? F RS) !#d*#&  #T z*#-#I#% "#W #$ !#X I-EP P r LATENT CONDITIONINGrb)denoisedrrsamplesamplingNc [R"U5n[R"UR5Ul[U5unnU b[R R U RSS5UUSU S9RSS5RSS5n [RRRX5Sn [U RS5H$n[U UR!S5U5U U'M& U bNU bKU R#S5n U Vs/sHn[%U5PM n n['U 5['U 5:XdS5eUc[)5RX4US5unOUnU bSU ;aU R+S5nU UUS-2SS2SS2SS24nUR-U5nUUS SS2SS2SURS 24'S U;aj[.R0"SSUS RS SS4[.R2US R4S 9nS U - USS2SS2SURS 24'UUS 'O)S U - US SS2SS2SURS 24'US nOSn[75R9UU5unn[75R9UUU- 5un n!['U5S:a*[;SU5 [=5R?U UUUUS9un"n#U bJ[AX5H;un$n%U%S:XaM[BR"UUUUU$R!S5U%U S9unnnM= Ub![E5RGUUUUUUUS9unnnURIUU5 [;SU 5 [=5R?U UUU US9un&n'[JR"UUU'S9unnn'UU'S '['U!5S:a*[;SU!5 [=5R?U UUU!U'S9un"n'U'UU4$s snf)Nr bilinear)rSr,INumber of optional cond images must match number of optional cond indicessamples noise_mask)dtypedevice?z*Denoising with no conditioning on sigmas: rCrrArB latent_imagerrr=latentimage frame_idxrRr=rrruguiding_latent latent_idxrRz'Denoising with conditioning on sigmas: rrruzLDenoising with no conditioning but with classical i2v noise mask on sigmas: )&rrrrcomfyutilscommon_upscalemovedimclamp comfy_extrasnodes_ltLTXVPreprocessexecuterangeshaper unsqueezesplitintlenrindexencodetorchonesfloat32rpr get_sigmasprintrrdzipr rgenerate set_condsr )(selfr<r=r>r?r@rrArBrCrPrQrRrSrTrUoptional_negative_index_latentsoptional_negative_index optional_negative_index_strengthoptional_initialization_latentsguiding_start_stepguiding_end_steprrilatentsidx_0 encode_pixelstconditioning_latent_frames_mask high_sigmas rest_sigmas middle_sigmas low_sigmas_ new_latents cond_imagecond_idxoutput_latentsdenoised_output_latentss( rrdLTXVBaseSampler.samples06" $ f.C.C D7?(  + **(00Q7 +Bq! !$0#8#8#G#G#O#O$$$ /55a89*7(+55a8$+$Q': ! ,1E1Q$9$?$?$D !5J$K5JSV5J !$K,-$2 [Z [ + 2-/77zSTUJW5G  +5J0J)//2E01BAq"1"1LMM =)A56GI q!\qwwqz\1 27*27**79-33A61=--"9-443/ GJHn/1l l0BC(G %>% ! z { a  > L46=="$ >  Q  +(+$)$ Hq= !((%%"$..q1&% )( + 6 #$--!!>29.   8, 7G4I4K4R4R  5S5 107E6L6L*7 3(3 1P - z?Q  ^ ,A+B+I+I!4 ,J, (Q'(8<L7LHH""   (, "),(,-g=r LTXVExtendSamplerc\\rSrSr\S5rSrSrSrSr S Sjr S r g) ri*cSSS04SSS04SSS04SS S [RS S S .4SSSSSSS .4SSS04SSS04SSS04SSS04SSSSSS.4S . SSS!04S"SS#04S$SSSSS%S.4S&.S'.$)(Nr&r'r(r)r*raz#The latents of the video to extend.r+Prgr zbIf -1, the number of frames will be based on the number of frames in the optional_guiding_latents.r.r/r0r1r'r3z]The overlap region to use for conditioning the new frames on the end of the provided latents.r4r5r6r7r8r9r:r;rG?rqzaThe strength of the conditioning on the overlapping latents, when using optional_guiding_latents.rJ) r<r=rnum_new_frames frame_overlaprrArBrCrRz'Optional latents to guide the sampling.rDrE)rFrI)optional_guiding_latentsrPrQcond_image_strengthrVrYr\s rr^LTXVExtendSampler.INPUT_TYPES.s1"I/L#MN +<=> EF #%!$33 !$H  ##%!" !$C  " QR& 3H'IJ#i1E%FG!I/S#TU#&""$G E+\ IJ-  RS)*5#&""#T ([A A r r`denoised_videorrrdreNc  [R"U5n[R"UR5UlUbNUbKURS5nUVs/sHn[ U5PM nn[ U5[ U5:XdS5e[ U5unnUSnURunnnn n!URun"n#n$UU"-n%US:XaU bU SRSU%- U"-n[5RUU%*S5un&Uc'[R"U!U#-U U$-U%U"-U-SS9Sn'OUn'U&SRU'SR5U&S'[5R!UUUU'U&SU S 9unnn'[#5R%UU5un(n)[#5R%U)UU- 5un*n+Ubh['S 5 [)X5HNun,n-U bU-S -S:Xa[+S U-S 35e[,R"UUUU'U,R/S5U-U S9unnn'MP [ U(5S:a<UR1UU5 ['SU(5 [35R5U UUU(U'S9un.n'U bO[5RU U%S5Sn [5R!UUUU'U U&SRSU S 9unnn'Ub![5R!UUUU'UUUS 9unnn'UR1UU5 ['SU*5 [35R5U UUU*U'S9un/n0[6R"UUU0S9unnn0[ U+5S:a[5R!UUUU0U&SU S 9unnn0Ubh['S 5 [)X5HNun,n-U bU-S -S:Xa[+S U-S 35e[,R"UUUU0U,R/S5U-U S9unnn0MP UR1UU5 ['SU+5 [35R5U UUU+U0S9un.n0[6R"UUU0S9unnn0Ub[95R;U0UUUS9Sn0[5RU0SS5Sn1[=5R?UU1U%S- SS9unUUU4$s snf)Nrirjrlrgrmr r>r?length batch_sizerrx Adding conditioning on keyframesr3Conditioning image index z (relative to the in the temporal chunk) is a multiple of 8 + 1,and guiding latents are used. Please provide other conditioning image indicesrtz4Denoising with overlap conditioning only on sigmas: rr,Denoising with full conditioning on sigmas: r{z@Denoising with overlap + keyframes conditioning only on sigmas: )r referencefactor per_frame)axis) rrrrrrrrdownscale_index_formularselect_latentsrrtorprrrrrrrr rrrrdr r batch_normalizeLinearOverlapLatentTransitionprocess)2rr<r=rrrrrArBrCrRguiding_strengthrrrPrQoptional_reference_latentsr adain_factorrrrrrnormalize_per_framerrrrlbatchchannelsframesr?r>time_scale_factorwidth_scale_factorheight_scale_factoroverlaplast_overlap_latentsrrrrrrrrrr!truncated_denoised_output_latentss2 rrdLTXVExtendSampler.samplexs166" $ f.C.C D ,1E1Q$9$?$?$D !5J$K5JSV5J !$K,-$2 [Z [8?()$18.x  ' ' C-/B #44 R $<$H(399!y*I*L*L  " ) )+ Y'  ) )/ *    $/=#;#;FDV#W [$/M$<$< ),>>% ! z + 4 5(+$)$ H,7HqLAhh!((%%&$..q1&0 ), { a    Xx 0 H+ V46=="( >  Q  $ /'8':'I'I('2(( $#$--!!"7/ :@@C).   + 6 #$--!!">29.   8, ,?,N,N #Q- - ) 34<< 6! != 8,,q%LsR r)rrqrqNNNNNrNrgrqrrFrrr rrr*sdB B H>L=LHH!%!"#'(,(, "),!3x-r LTXVInContextSamplercT\rSrSr\S5rSrSrSrSr S Sjr S r g) risc hSSS04SSS04SSS04SSS 04S SS 04S SS 04S.SSS04SSSSSSS.4S.S.$)Nr)r'r*r4r5r6r7r8r9r:r;razGThe latents to use for guiding the sampling, typically with an IC-LoRA.)r=rrArBrCguiding_latentsrDzThe image to use for conditioning the sampling, if not provided, the sampling will be unconditioned (t2v setup). The image will be resized to the size of the first frame.r+rgrr zYIf -1, the number of frames will be based on the number of frames in the guiding_latents.r)rPr@rVrr\s rr^ LTXVInContextSampler.INPUT_TYPESxs +<=> QR& 3H'IJ#i1E%FG!I/S#TU!#l$$!$P)#%!# !#~  ## # r r`rrdreNc \[R"U5n[R"UR5UlUcSnUbNUbKURS5nUVs/sHn[ U5PM nn[ U5[ U5:XdS5e[ U5unnURunnnUSRunnnnnU S:wa U S- U-S-nU bU nO)[R"UU-UU-US- U-S-SS9Sn[5RUU5unn[5RUUU- 5un n![ U5S:a*[SU5 [5RUUUUUS 9un"nUb'SU;a![!5R#USS5SnS n#OS n#[S 5 [%5R'UUUUUU#(aSOSUS 9unnnUbf[SU5 [)Xx5HKun$n%U%S-S:Xa[+SU%S35e[,R"UUUUU$R/S5U%US9unnnMM U b![%5R'UUUUU U U S 9unnnUR1UU5 [SU 5 [5RUUUU US 9un"n&[2R"UUU&S9unnn&[ U!5S:aVUR1UU5 [SU!5 [5RUUUU!U&S 9un"n&[2R"UUU&S9unnn&U&UU4$s snf)Nrirjrlrgr rrz8Denoising with keyframes only [if available] on sigmas: rrTFz&Adding conditioning on guiding latentsrxrr3rz] is a multiple of 8 + 1 and guiding latents are used. Please provide other cond image indicesrtrr{zEDenoising with keyframes only [if available] conditioning on sigmas: )rrrrrrrrrrrrrrrrdrrrrrrr rrr )'rr=rrArBrCrrPrQr@rrrrrrrrrrrrrrrrrr?r>rrrrrrskip_one_guiding_latentrrrs' rrdLTXVInContextSampler.samples>(6" $ f.C.C D  '$( ! ,1E1Q$9$?$?$D !5J$K5JSV5J !$K,-$2 [Z [8?(  ' ' C-/B2A1K1Q1Q.x   1n)::Q>F * 69K.6600 33 &77!;   K$/=#;#;FDV#W [$/M$<$< ),>>% ! z { a  J  56=="( >  Q  ! ,6K1K/1@@BO'+ #&+ # 67  ) )*3q% *     + 46K L(+$)$ Ha<1$$3H:>[\!((%%&$..q1&0 )( + 6 #$--!!">29.   8, 'E'E $ (F( $ #7E6L6L*7 3(3 z?Q    Xx 0 W ,A+B+I+I!4 ,J, (Q';I:P:P!!.; 7Hh 7 (8<L=LHH""(,(, "),%g=r zLinear transition with overlap) descriptionc@\rSrSr\S5rSrSrSrSr S Sjr Sr g ) riNc,SSSSSSS.4S.SSSS 040S .$) Nrar+r )r.r/r0)samples1samples2rrr.rrVrr\s rr^)LinearOverlapLatentTransition.INPUT_TYPESPs=('!q3#GH A/  r rrzLightricks/latentc0UR5nUSnU$)Nrl)r)rrlr]s r get_subbatch*LinearOverlapLatentTransition.get_subbatchbs LLNI,r cURU5nURU5n[R"SSUS-5SSRUR5nS/UR 5-nUR S5Xd'URU5n[S5/UR 5-nUR5n[U*S5X'UR5n [SU5X'UR5n [SU*5X'UR5n [US5X'U[U 5XQ[U5-SU- U[U 5--U[U 5/n [R"XS9n [R"SU RS5R[RS9nU US.4$)Nr rrmrg)dim)ro)rl batch_index)rrlinspacerrprsizereshapeslicertuplecatarangerr)rrrrralphar slice_allslice_overlap1slice_overlap2 slice_rest1 slice_rest2partscombined_samplescombined_batch_indexs rr%LinearOverlapLatentTransition.processgs$$X.$$X.q!Wq[1!B7::8??Khlln$jjm  e$4[MHLLN2 ")$gXt4")$Q0nn& !$1 nn& !'40  U;' ( U>23 35yHU>%:;; < U;' (  !99U5$||A/?/E/Ea/HILL-- M  ,3   r rN)r) rrrrrr^rrrrrrrr rrrNs/    LH"H ) r rzLTXV Normalizing SamplercP\rSrSr\S5r\S\R4Sj5rSr g)LTXVNormalizingSampleric[R"SS[RRS5[RRS5[R RS5[R RS5[RRS5[RRSS S 9[RRS S S 9/[RRS S9/S9$)NrutilityrCrrArBrsvideo_normalization_factorsz1,1,1,1,1,1,1,1)r.audio_normalization_factorsz1,1,0.25,1,1,0.25,1,1denoised_output) display_name)node_idcategoryinputsoutputs) rSchemaNoiseInputGuiderSamplerSigmasLatentStringOutput)clss r define_schema$LTXVNormalizingSampler.define_schemasyy,w' )   + ) / 1;L  1;R     .? @!  r returnc hURRRRRS:wa [ 5eURRRnUR S5nUR S5nUV s/sHn [U 5PM nn UV s/sHn [U 5PM nn [U5S- n [U5U :a3[U5S:a$URUS/U [U5- -5 [U5U :a3[U5S:a$URUS/U [U5- -5 [[Xg55V V V s/sHun upU S:wdU S:wdMU S-PM nn n n [SU-SS 9 S nU"XN5n[S U-SS 9 Sn UHnU [U5S- - n [S U-SS 9 [5RUUUUUS 9unnURUSR S5unnU S- [U5:dMuU S- [U5:dMUXkS- -nUX{S- -n[#UR%UU55US'[SU XkS- X{S- 4-SS 9 M [&R("U5$s sn fs sn fs sn n n f)Nrrir rrgrqzSampling split indices: %sT)flushcU(dU/$[[U55n/nSnUH nXE:aURXUS-5 UnM" U[U5:aURXS5 U$)z Splits arr into chunks according to indices (split points). Indices are treated as starting a new chunk at each index in the list. rr N)sortedsetappendr)arrindices split_pointschunksprevidxs rsplit_by_indices8LTXVNormalizingSampler.execute..split_by_indicessq u !#g,/LFD#:MM#S1W"56$c#h c%j)Mr zSigmas chunks: %szSampling with sigmas %srrrlz )L \"Q& &A +| J-rN) rrrrrr&rr@rrrr rrrs9  ,n+ n+n+r r)"rr|rrZrcomfy.ldm.lightricks.av_modelrcomfy.nested_tensorrcomfy_api.latestr!comfy_extras.nodes_custom_samplerrrcomfy_extras.nodes_ltrr r guider latent_normr rrrnodes_registryrrr!rrr ComfyNoderrr rrTs  4,PTT (:&  B=B=B=J C-C-C-L  U=U=U=p 89A A :A H 23G+R\\G+4G+r