
    
3j                     B    S SK J r   SSKJr  \" SS9 " S S5      5       rg)	    )copy   )
comfy_nodezSet VAE Decoder Noise)namec                   6    \ rS rSr\S 5       rSrSrSrS r	Sr
g)	DecoderNoise   c           
      B    SSSSSSSSS	.4SS
SSSSS	.4SSSSSS.4S.0$ )NrequiredVAEFLOATg?g        g      ?gMbP?z)The timestep used for decoding the noise.)defaultminmaxsteptooltipg?z,The scale of the noise added to the decoder.INT*   r   l    z,The random seed used for creating the noise.)r   r   r   r   )vaetimestepscaleseed )clss    K/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/decoder_noise.pyINPUT_TYPESDecoderNoise.INPUT_TYPES   sn     #'"" %#N	 #("" %#Q	 #% 1#Q	-!
 !	
    	add_noiser   zlightricks/LTXVc                     [        U5      n[        US5      (       a   X%R                  l        X5R                  l        X%l        X5l        XEl        U4$ )Nfirst_stage_model)r   hasattrr"   decode_timestepdecode_noise_scale_decode_timestepr   )selfr   r   r   r   results         r   r    DecoderNoise.add_noise1   sJ    c6.//7?$$4:?$$7"*$)!yr   r   N)__name__
__module____qualname____firstlineno__classmethodr   FUNCTIONRETURN_TYPESCATEGORYr    __static_attributes__r   r   r   r   r      s+    "
 "
H HL Hr   r   N)r   nodes_registryr   r   r   r   r   <module>r4      s*     & ()2 2 *2r   