o >g @sJddlmZddlZddlZGdddZGdddejZeedZdS))UNION_CONTROLNET_TYPESNc@s,eZdZeddZdZdZdZddZdS) SetUnionControlNetTypecCsdddgttfdiS)Nrequired CONTROL_NETauto) control_nettype)listrkeyssrG/home/comdoleger1/zen-flux-style-shape/comfy_extras/nodes_controlnet.py INPUT_TYPESsz"SetUnionControlNetType.INPUT_TYPESconditioning/controlnetrset_controlnet_typecCsB|}t|d}|dkr|d|g|fS|dg|fS)Nr control_type)copyrget set_extra_arg)selfrr Z type_numberrrrrs  z*SetUnionControlNetType.set_controlnet_typeN) __name__ __module__ __qualname__ classmethodrCATEGORY RETURN_TYPESFUNCTIONrrrrrrs  rc@s(eZdZeddZdZdZddZdS) ControlNetInpaintingAliMamaApplycCsFdddddddddd d d d fdd d dd d fddd dd d fd iS)Nr) CONDITIONINGr)VAE)IMAGE)MASKFLOAT?gg$@g{Gz?)defaultminmaxstepgMbP?) positivenegativervaeimagemaskstrength start_percent end_percentrr rrrrsz,ControlNetInpaintingAliMamaApply.INPUT_TYPESapply_inpaint_controlnetrc Csg} |jrrControlNetApplyAdvancedr NODE_CLASS_MAPPINGSrrrrs