3junSSKJr SSKrSSKJr SSKrSSKrSSK J r SSK J r SSK Jr \"SS9"S S55r\"S S9"S S 55r\"S S9"S S 55r\"SS9"SS55r\"SS9"SS55r\"SS9"SS55rSSjr\"SS9"SS55rg))OptionalN)LATENT_DOWNSAMPLE_FACTOR) NestedTensor) comfy_nodeLTXVSelectLatents)namecR\rSrSrSr\S5rSrSrSr Sr S\ S \ S \ S \ 4S jrS rg)r z Selects a range of frames from a video latent. Features: - Supports positive and negative indexing - Preserves batch processing capabilities - Handles noise masks if present - Maintains 5D tensor format c .SSSSSSSS.4SS SSSS.4S .0$) NrequiredLATENTINTr'rdefaultminmaxstep)samples start_index end_indexss E/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/latents.py INPUT_TYPESLTXVSelectLatents.INPUT_TYPESsC & !%aH   "5qI   rselect_latents latent/videozSelects a range of frames from the video latent. start_index and end_index define a closed interval (inclusive of both endpoints).rrrreturncUR5nUSnURupgpn US:aX-OUn US:aX-OUn [S[XS- 55n [S[XS- 55n X:a [X5n USS2SS2XS-2SS2SS24US'SU;a"USbUSSS2SS2XS-2SS2SS24US'U4$![an [ S[ U 535 eSn A ff=f)aq Selects a range of frames from the video latent. Args: samples (dict): Video latent dictionary start_index (int): Starting frame index (supports negative indexing) end_index (int): Ending frame index (supports negative indexing) Returns: tuple: Contains modified latent dictionary with selected frames Raises: ValueError: If indices are invalid rrrN noise_maskzTXVSelectLatents] Error: )copyshaperr Exceptionprintstr)selfrrrr video_latentbatchchannelsframesheightwidth start_idxend_idxes rr# LTXVSelectLatents.select_latents1s  AYr?r@rArUTensorrMrTrCrrWrErr"rrFrF]s  LHH _ ) )05 ) )VELL5<<8   05  GJ UX  %,,  r"LTXVSetVideoLatentNoiseMasksc $\rSrSrSr\S5rSrSrSr Sr S\ S \ RS \4S jrS \ RS \ R4S jrS \ RS \S\S \ R4SjrS\ RS\ RS\S\S S4 SjrSrg)rwa Applies multiple masks to a video latent. Features: - Supports multiple input mask formats (2D, 3D, 4D) - Automatically handles fewer masks than frames by reusing the last mask - Resizes masks to match latent dimensions - Preserves batch processing capabilities Input Formats: - 2D mask: Single mask [H, W] - 3D mask: Multiple masks [M, H, W] - 4D mask: Multiple masks with channels [M, C, H, W] cSSSS.0$)Nr r)MASK)rmasksrrs rr (LTXVSetVideoLatentNoiseMasks.INPUT_TYPESs &"  r"rset_maskr$zApplies multiple masks to a video latent. masks can be 2D, 3D, or 4D tensors. If there are fewer masks than frames, the last mask will be reused.rr|r%c:UR5nUSnURupVpxn SU;a2[R"USXxU 4URUR S9US'OIUSRn U SU:wa1[R"USXxU 4URUR S9US'UR U5n U RSn URXU 5n URUSXU 5 U4$![an[S[U535 eSnAff=f) a Applies masks to video latent frames. Args: samples (dict): Video latent dictionary containing 'samples' tensor masks (torch.Tensor): Mask tensor in various possible formats - 2D: [H, W] single mask - 3D: [M, H, W] multiple masks - 4D: [M, C, H, W] multiple masks with channels Returns: tuple: Contains modified latent dictionary with applied masks Raises: ValueError: If mask dimensions are unsupported RuntimeError: If tensor operations fail rr'rdtyperRrOrz&[LTXVSetVideoLatentNoiseMasks] Error: N) r(r)rUrtrrR_reshape_masks _resize_masks _apply_masksr*r+r,)r-rr|rr. batch_sizer0 num_framesr2r3existing_shapemasks_reshapedM resized_masksr6s rr~%LTXVSetVideoLatentNoiseMasks.set_masks/$!  AYJ>P>P ;J*e1$"'++J>&,,'..#, "#<!6!6!!$ 2&+kk#Q EB*00+22'AlO"007N$$Q'A ..~uMM   a o}! L4K  :3q6(C D  sC0C33 D=DDc[UR5nURnUS:Xa URS5RS5$US:Xa;UR URSSURSURS5$US:Xa;UR URSSURSURS5$[ SUS35e)z-Reshapes input masks to consistent 4D format.rOrrr`zUnsupported 'masks' dimension: z0. Must be 2D (H,W), 3D (M,H,W), or 4D (M,C,H,W).)rDr)ra unsqueezereshaperb)r-r|original_shapendimss rr+LTXVSetVideoLatentNoiseMasks._reshape_masks(su{{+  A:??1%//2 2 aZ==QEKKNEKKPQNS S aZ==QEKKNEKKPQNS S1.1ABAA r"r2r3cX[RRRXU4SSS9$)z-Resizes all masks to match latent dimensions.bilinearFsizemode align_corners)rUnn functional interpolate)r-r|r2r3s rr*LTXVSetVideoLatentNoiseMasks._resize_masks9s1xx"".. j/  r"r'rrrNc`[U5Hn[XTS- 5nX&USS2SS2U4'M! g)z$Applies resized masks to all frames.rN)ranger)r-r'rrrfmask_idxs rr)LTXVSetVideoLatentNoiseMasks._apply_masksAs3z"A1!e}H"/"9Jq!Qw #r"r)r8r9r:r;r<r=r r>r?r@rArBrUrvrDr~rrCrrrErr"rrwrws   LHH N 33U\\3e3jELLU\\" \\ +. 7:   :LL :|| : :  :  :r"LTXVDilateLatentcN\rSrSr\S5rSrSrSrSr S\ S\ S \ S \ 4S jr S rg )riNc .SSSSSSSS.4SSSSSS.4S.0$)Nr rrrdr)latenthorizontal_scalevertical_scalerrs rr LTXVDilateLatent.INPUT_TYPESPsC % !!CC%  !!CC#   r"r dilate_latentr$z Dilates a latent by a grid size.rrrr%c(US:Xa US:XaU4$USnURSS5nURSSURSU-URSU-4-n[R"UURUR SS9nXGSSSU2SSU24'URS SURS URSURS4n[R "US URUR SS9n UbUOS U SSSU2SSU24'XyS .nU4$)Nrrr'rr`F)rRr requires_grad.rrOg?rr')getr)rUrtrRrfull) r-rrrrmask dilated_shapedilated_samplesdilated_mask_shape dilated_masks rrLTXVDilateLatent.dilate_latentes^ q ^q%89 #zz,- bq) MM! ~ - MM! / /-   ++ >>--   FM-~-/A1A/AAB  ! !! $  ! !! $  ! !! $  ! !! $  zz  >>--  $D# S*N*,>.>,>>?-Iyr"rN)r8r9r:r;r=r r>r?r@rArBrCrDrrErr"rrrNsO    LHH4K((.1(CF( (r"LTXVAddLatentGuidec>\rSrSr\S5rSrSrSrSr Sr Sr S r g ) ric 6SSSSSSSSSSS S S .4S S SS S.4S.0$)Nr VAE CONDITIONINGrrrrrrzLatent index to start the conditioning at. Can be negative toindicate that the conditioning is on the frames before the latent.)rrrrtooltipFLOATrrrr)vaepositivenegativerguiding_latent latent_idxstrengthrrs rr LTXVAddLatentGuide.INPUT_TYPESsU --%"-#$$# !$]  %#c#&NO#  r")rrr)rrr ltxtricksgeneratez=Adds a keyframe or a video segment at a specific frame index.c "[R"U5nUSnUSn [U RSS5n URSU RS-S:Xa#URSU RS-S:XdS5e[ 5R UURSU RS-URSU RS-S9SnUSn UR SS5n U RSU RS-U RS-n URn US::aXmS-nOS US - U S--n[RRUUUUUU UU U S 9 up#pHS S K J n U"X,U 5nU"X!<<?ekk!n<:    y)#'' d; $kk!nu{{1~= AN33 ?"1%55IZ!^}Q/???I191F1F1V1V! '!2W 2 .F C/ +; 0 +;    9  r"rN) r8r9r:r;r=r r> RETURN_NAMESr@r?rArrErr"rrrs4  .>L5LHHQK= r"LTXVImgToVideoConditionOnlycL\rSrSrSr\S5rSrSrSr Sr Sr S r S S jr S rg )riz Applies image conditioning to the first frames of a video latent. Features: - Takes existing latent and applies image conditioning - Automatically resizes image to match latent dimensions - Creates noise mask for strength control - Supports bypass mode c0SSSSSSSS.4S.S S S S S .40S.$)Nr)IMAGErrrrr)rimagerrbypassBOOLEANFzBypass the conditioning.rrr optionalrrs rr 'LTXVImgToVideoConditionOnly.INPUT_TYPESsF #%$#c#&NO  %2LM  r"r)rzconditioning/video_modelsrz|Applies image conditioning to the first frames of an existing latent. Creates a noise mask to control conditioning strength.cZURupEnUupxpn X-n X-n URSU :wdURSU :waC[RR UR SS5XSS5R SS5n OUn U SS2SS2SS2SS24n UR U 5nU$)NrrOrrcenterr)rr)comfyutilscommon_upscalemovedimencode)r-rr)rtime_scale_factorheight_scale_factorwidth_scale_factorr/r0r1r2r3pixels encode_pixelsts r encode_image(LTXVImgToVideoConditionOnly.encode_images  ' ' C0B27. #% ;;q>V #u{{1~'>[[// b!$eZgan Fq!Q{+ JJ} %r"cRU(aU4$USnURXRU5nXvSS2SS2SURS24'[R"SSURSSS4[RUR S9nSU- USS2SS2SURS24'XhS.4$)NrrOrrrr)rr)rUonesfloat32rR) r-rrrrrrrconditioning_latent_frames_masks rr$LTXVImgToVideoConditionOnly.generate.s 9 #   e]]C 8&'1l l"#+0** 7==#Q *-->>+ ' ?BHn'1l l(:;#SUUr"rN)F)r8r9r:r;r<r=r r>rr@r?rArrrErr"rrrsG  LL*HH A ,Vr"c S/U-n[S[US- U55n[U[US- U55n[SU5n[SX- 5n[XP5HnXe- S-U- XF'M [XS-5HnSXF'M [X!U-S-5n[US-U5H%nSXaS-- S-U- - XF'[SXF5XF'M' SSKn[ U5n U S- S-S-n UR "S/[ [SU S55-5n UR "U5n UR"U 5n UR"XU 5R5nXN4$)a Returns a blend coefficient list shaped such that: - It is 0.0 outside the range [video_frame_index_start, video_frame_index_end] - It ramps up from 0.0 to 1.0 starting at video_frame_index_start-slope_len to video_frame_index_start - It stays at 1.0 during [video_frame_index_start, video_frame_index_end] - It ramps down from 1.0 to 0.0 starting at video_frame_index_end to video_frame_index_end+slope_len - The 'slope_len' parameter controls the sharpness (in frames) of the slope. rrrrN) rrrnumpylenarrayrarangeinterptolist)video_frame_index_startvideo_frame_index_endvideo_frame_count slope_lencoeffs rampl_starti rampr_endnp num_coeffspixel_frame_lengthxpfppixel_frame_positionspixel_frame_coefficientss r#get_video_latent_blend_coefficientsrBsU& &F" 3 1$&= > %6%:B & BII&89!yy)>BGNNP  ++r"z!LTXV Set Audio Video Mask By Time) descriptioncD\rSrSr\S5rSrSrSrSr Sr S S jr S r g) LTXVSetAudioVideoMaskByTimei{cSSSSSSSSSSS.4SS SSS.4SS SS S.4S S S04S S S04SSSSS.4SSSSS.4SSSSSS.4S.SSSSS.40S.$)Nrr)MODELrrrg@@rg$@g8@g@@rrTrrrrrr) av_latentrrmodelr audio_vae start_timeend_time video_fps mask_video mask_audiomask_init_value_videomask_init_value_audior spatial_maskr{z Spatial mask.rrr)clss rr 'LTXVSetAudioVideoMaskByTime.INPUT_TYPES}s)--#%&CF(ST$$s6&RS%4E'RS(9d*;<(9d*;< #C<*  #C<*  !!CC)4 $A!5  r")rrrrr)rrrvideo_latent_blend_coefficientsvideo_pixel_blend_coefficientsrunutilityz&Sets the audio and video mask by time.Nc SSKJn URRRR S:wa [ 5eURRnURRnURRnUU- [- nURSnU U- n[US[5(d [ S5eURUSRS5unn[ R""UR$SUR$SUR$SUR$SUR$S 4U S 9n[ R""UR$SUR$SUR$SUR$S4U S 9nUbUR&S:XaUR)S5nUR&S:Xa UR)S5R)S5n[ R*R,R/UUR$SUR$S 4S S S 9nUR$SnUR$SnUS- U-S-n[0R2"S/[5[7SUU-U55-5n[9[;Xy-55n[0R<"UUSS9n [9[;X-55n![0R<"UU!SS9S- n"[9[;UU-55n#[9[;UU-55S-n$[?SU 5n [AU"U5n"[?SU#5n#[AU$U5n$[CSU U"U#U$U UUUUUU!UU4 -5 U (a&UbUUSS2SS2U U"2SS2SS24'OSUSS2SS2U U"24'U (aSUSS2SS2U#U$24'SU;aUSRSRE5n%U%R$SU%R$Ss=:Xa.Ss=:Xa'U%R$Ss=:XaU%R$S :XaDO OA[7U%R$S5H%n&USS2SS2U&SS2SS24==U%SSU&SS4-ss'M' [URGUU55US'[IU U"UUS9un'n(UUUU'U(4$)Nr) LTXAVModelr+rz av_latent must be a NestedTensorrrOrr`) fill_valuerFrleft)siderighta"noise mask start and end indices: video (%d %d), audio (%d %d), video fps: %f, video_latents_per_second: %f, audio_latents_per_second: %f, video_latent_frame_count: %d, video_pixel_frame_count: %d, video_pixel_frame_start_raw: %d, video_pixel_frame_end_raw: %d, start_time: %f, end_time: %frr')r)%comfy.ldm.lightricks.av_modelr+rdiffusion_model __class__r8rb autoencoder sampling_ratemel_hop_lengthrr isinstancer separate_audio_and_video_latentstensorsrUrr)rarrrrr rrrrCround searchsortedrrr+clone!recombine_audio_and_video_latentsr))r-rrrrrrrrrrr r!r"rr#r+ltxavr4r5audio_latents_per_secondrvideo_latents_per_second video_samples audio_samples video_mask audio_maskvideo_latent_frame_countaudio_latent_frame_countvideo_pixel_frame_countrvideo_pixel_frame_start_rawvideo_latent_frame_index_startvideo_pixel_frame_end_rawvideo_latent_frame_index_endaudio_latent_frame_index_startaudio_latent_frame_index_end base_maskframer&r's) rr(LTXVSetAudioVideoMaskByTime.runs8$ = ;; & & 0 0 9 9\ I, !KK77"--;; "..== N *-E E ! 77:#,/@#@ )I. ==?@ @','M'M i ( ( ( $ }ZZ##A&##A&##A&##A&##A&  -  ZZ##A&##A&##A&##A&   -   #  A%+55a8   A%+55a8BB1E  88..::#))!,m.A.A!.DE# ;L $1#6#6q#9 #0#6#6q#9 #;a#?CT"TWX"X XX Ca03DDFWX   '*% 0F*G&H#)+ +&* &%(h.B(C$D! OOB 9 H1 L %*- *77 8* & h!99: ;a ? %*-Q0N)O&'* (*B( $*-Q0N)O&'* (*B( $  f/,.,((('+)  & '! 23OO q89UUU  145QQQ  9 $!,/77:@@BI"??1%&&??1%&??1% & #9??1#56Eq!UAq01Yq!UAq?P5QQ17#/  3 3J K# , 0.,(#   H')G    + *   r"r)N) r8r9r:r;r=r r>rr?r@rAr(rErr"rrr{s@! ! FPLLHH:K$!m r"r)r)typingr comfy.utilsrcomfy_extras.nodes_ltrrr rU"comfy.ldm.lightricks.vae.audio_vaercomfy.nested_tensorrnodes_registryrrrFrwrrrrrrr"rrVs( G,& $%LL&L^ !"oo#od /0z:z:1z:z #$>>%>B %&^ ^ '^ B ./LVLV0LVf 6,r ;<_ _ =_ r"