
    +j]                         d dl mZ d dlZd dlZd dlZd dlmZmZ  G d dej	                  Z
 G d de          Zdefd	ZdS )
    )overrideN)ComfyExtensionioc                   L    e Zd Zed             Zeddej        fd            ZdS )EmptyMochiLatentVideoc                    t          j        ddt           j                            dddt          j        d          t           j                            dddt          j        d          t           j                            d	d
dt          j        d          t           j                            dddd          gt           j                                        g          S )Nr   zmodel/latent/video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_schemaz#EmptyMochiLatentVideo.define_schema	   s    y+)Wcru?SZ\]]Xs@T[]^^Xrqe>RYZ[[\1!FF	 	  ""
 
 
 	
    r   returnc                     t          j        |d|dz
  dz  dz   |dz  |dz  gt          j                                                  }t          j        d|i          S )N   r   r      )devicesamples)torchzeroscomfymodel_managementintermediate_devicer   
NodeOutput)r!   r	   r   r   r   latents         r"   executezEmptyMochiLatentVideo.execute   sy    j"
q/@A.EvQR{TY]^T^_hmh~  iS  iS  iU  iU  V  V  V}i0111r$   N)r   )__name__
__module____qualname__classmethodr#   r   r0   r2    r$   r"   r   r      s[        
 
 [
 2 2R] 2 2 2 [2 2 2r$   r   c                   L    e Zd Zedeeej                          fd            ZdS )MochiExtensionr%   c                    K   t           gS N)r   )selfs    r"   get_node_listzMochiExtension.get_node_list    s       "
 	
r$   N)	r3   r4   r5   r   listtyper   	ComfyNoder=   r7   r$   r"   r9   r9      sE        
T$r|*<%= 
 
 
 X
 
 
r$   r9   r%   c                  "   K   t                      S r;   )r9   r7   r$   r"   comfy_entrypointrB   '   s      r$   )typing_extensionsr   r+   comfy.model_managementr-   r   comfy_api.latestr   r   r@   r   r9   rB   r7   r$   r"   <module>rF      s    & & & & & &       / / / / / / / /2 2 2 2 2BL 2 2 2.
 
 
 
 
^ 
 
 
      r$   