
    
3jV                     B    S SK Jr  SSKJr  \" SS9 " S S5      5       rg)    
dtype_size   )
comfy_nodeLTXVPatcherVAE)namec                   :    \ rS rSr\S 5       rSrSrSrSr	S r
Srg	)
r      c                     SSS00$ )NrequiredvaeVAE )ss    I/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/vae_patcher.pyINPUT_TYPESLTXVPatcherVAE.INPUT_TYPES   s     x
 	
    r   patchzlightricks/LTXVzLTXV VAE Patcherc                 J    SSK Jn  UR                  nS Ul        U" USS9  U4$ )Nr   )	patch_vaec                 `    U S   U S   -  U S   -  U S   -  S-  S-  S-  S-  [        U5      -  $ )Nr               r   )shapedtypes     r   <lambda>&LTXVPatcherVAE.patch.<locals>.<lambda>   se    qAh"Ah" Ah" 	"
 " " " " r   r   )patch_block) q8_kernels.integration.patch_vaer   first_stage_modelmemory_used_decode)selfr   r   	vae_models       r   r   LTXVPatcherVAE.patch   s0    >))	  	 	)+vr   r   N)__name__
__module____qualname____firstlineno__classmethodr   RETURN_TYPESFUNCTIONCATEGORYTITLEr   __static_attributes__r   r   r   r   r      s/    
 
 LH HEr   N)comfy.model_managementr   nodes_registryr   r   r   r   r   <module>r5      s*    - & !"  #r   