
    
3j                     >    S SK r SSKJr  \" SS9 " S S5      5       rg)    N   )
comfy_nodeDynamicConditioning)namec                       \ rS rSr\S 5       rSrSrSrSr	S r
S\R                  S	\R                  4S
 jrS\R                  S\R                  S\4S jrSrg)r      c                 (    SSSSSSSS.4S	S
S04S.0$ )NrequiredMODELFLOATg?r      g{Gz?)defaultminmaxstepBOOLEANr   T)modelpoweronly_first_frame )ss    R/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/dynamic_conditioning.pyINPUT_TYPESDynamicConditioning.INPUT_TYPES   s8     #!s1QPT#UV%.D0A$B
 	
    r   applyzlightricks/LTXVFc                 v    X0l         X l        UR                  5       nUR                  U R                  5        U4$ )N)r   r   cloneset_model_denoise_mask_functionforward)selfr   r   r   s       r   r   DynamicConditioning.apply   s1     0
--dll;xr   sigmastep_sigmasc                 X    [        U5       H  u  p4XA::  d  M  Us  $    [        U5      S-
  $ )Nr   )	enumeratelen)r"   r$   r%   i
step_sigmas        r   	find_stepDynamicConditioning.find_step   s0    &{3MA" 4 ;!##r   denoise_maskextra_optionsc                    US   nUS   nU R                  X5      nU R                  U-  nUR                  5       nU R                  (       aA  UR                  R
                  R                  R                  nUS S 2S U2S S24==   U-  ss'   OX'-  nUR                   HF  n	SU	;   d  SU	;   d  M  UR                  U	    H"  n
SU
;   d  M  SU
S   ;   d  M  X*S   S   l	        M$     MH     U$ )Nr   sigmasr   positivenegativemodel_condsr-   )
r+   r   r   r   model_patcherr   diffusion_modelin_channelscondscond)r"   r$   r-   r.   r   r%   r   r   num_channelskr8   s              r   r!   DynamicConditioning.forward$   s     g&#H-~~e1

D #))+   ..44DDPPLM\M2A2-.58."LAQ*/!KKND$,4CV1VCO]+N;@ +  r   )r   r   N)__name__
__module____qualname____firstlineno__classmethodr   RETURN_TYPESFUNCTIONCATEGORYINITr   torchTensorr+   dictr!   __static_attributes__r   r   r   r   r      so    
 
 LH HD$u|| $%,, $\\16NRr   )rE   nodes_registryr   r   r   r   r   <module>rJ      s*     & &'2 2 (2r   