o ix@s&ddlZddlZejjZdddZdS)Nư>cCsN|durtjjj||jdf|dStjjj||jtjj||j|j d|dS)N)eps)dtypedevice)weightr) torchnn functionalrms_normshapecomfymodel_managementcast_torr)xrrr+/mnt/c/Users/fbmor/ComfyUI/comfy/rmsnorm.pyr s*r )Nr)rcomfy.model_managementr r RMSNormr rrrrs