o >g@s2ddlZddlZddlZGdddZdeiZdS)Nc@s.eZdZeddZdZdZdZd ddZd S) EmptyMochiLatentVideoc CsNddddtjddfdddtjddfdddtjd dfdd d d d fd iS)NrequiredINTiP)defaultminmaxstepii)rrr)widthheightlength batch_size)nodesMAX_RESOLUTION)srB/home/comdoleger1/zen-flux-style-shape/comfy_extras/nodes_mochi.py INPUT_TYPESs z!EmptyMochiLatentVideo.INPUT_TYPES)LATENTgeneratez latent/videor cCs<tj|d|ddd|d|dgtjd}d|ifS)N r r )devicesamples)torchzeroscomfymodel_managementintermediate_device)selfrrrrlatentrrrrs2 zEmptyMochiLatentVideo.generateN)r ) __name__ __module__ __qualname__ classmethodr RETURN_TYPESFUNCTIONCATEGORYrrrrrrs r)rrZcomfy.model_managementr rNODE_CLASS_MAPPINGSrrrrs