
    3j]                         S SK Jr  S SKrS SKrS SKrS SKJrJr   " S S\R                  5      r
 " S S\5      rS\4S	 jrg)
    )overrideN)ComfyExtensionioc                   T    \ rS rSr\S 5       r\SS\R                  4S jj5       rSr	g)EmptyMochiLatentVideo   c                    [         R                  " SS[         R                  R                  SSS[        R
                  SS9[         R                  R                  SSS[        R
                  SS9[         R                  R                  S	S
S[        R
                  SS9[         R                  R                  SSSSS9/[         R                  R                  5       /S9$ )Nr   zmodel/latent/mochiwidthiP     )defaultminmaxstepheighti  length         
batch_size   i   )r   r   r   )node_idcategoryinputsoutputs)r   SchemaIntInputnodesMAX_RESOLUTIONLatentOutput)clss    8/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_mochi.pydefine_schema#EmptyMochiLatentVideo.define_schema	   s    yy+)Wcru?S?SZ\]Xs@T@T[]^Xrqe>R>RYZ[\1!F	 		  "
 	
    returnc                     [         R                  " USUS-
  S-  S-   US-  US-  /[        R                  R	                  5       S9n[
        R                  " SU05      $ )N   r   r   r   )devicesamples)torchzeroscomfymodel_managementintermediate_devicer   
NodeOutput)r"   r
   r   r   r   latents         r#   executeEmptyMochiLatentVideo.execute   sj    j"
q/@A.EvQR{TY]^T^_hmh~h~  iS  iS  iU  V}}i011r&    N)r   )
__name__
__module____qualname____firstlineno__classmethodr$   r   r1   r3   __static_attributes__r5   r&   r#   r   r      s3    
 
 2R]] 2 2r&   r   c                   L    \ rS rSr\S\\\R                        4S j5       r	Sr
g)MochiExtension   r'   c                    #    [         /$ 7fN)r   )selfs    r#   get_node_listMochiExtension.get_node_list    s      "
 	
s   
r5   N)r6   r7   r8   r9   r   listtyper   	ComfyNoderB   r;   r5   r&   r#   r=   r=      s)    
T$r||*<%= 
 
r&   r=   r'   c                     #    [        5       $ 7fr@   )r=   r5   r&   r#   comfy_entrypointrH   '   s     s   )typing_extensionsr   r,   comfy.model_managementr.   r   comfy_api.latestr   r   rF   r   r=   rH   r5   r&   r#   <module>rL      s<    &    /2BLL 2.
^ 
 r&   