o i@sjddlmZddlZddlmZmZGdddejZGdddejZGdd d eZ d e fd d Z dS) )overrideN)ComfyExtensionioc@s,eZdZeddZedejfddZdS) RenormCFGc CsPtjddtjdtjjdddddd d tjjd d dddd d gtjgd S)Nrzadvanced/modelmodel cfg_truncdgY@g{Gz?T)defaultminmaxstepadvanced renorm_cfgg?)node_idcategoryinputsoutputs)rSchemaModelInputFloatOutputclsr8/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_lumina2.py define_schemas zRenormCFG.define_schemareturncs,fdd}}||t|S)Ncs|d}|d}|d}|d}|d}jjj}|dkr|ddd|f|ddd|f}}|dd|df|dd|df} } ||||} | } tdkrtjj|ttdt |j d d } | t}tjj| ttdt | j d d }||kr| ||} n.|ddd|f|ddd|f}}|dd|df|dd|df} } |} | } tj | | gdd }||S) N cond_denoiseduncond_denoised cond_scaletimestepinputrr T)dimkeepdim)r%) rdiffusion_model in_channelsfloattorchlinalg vector_normtuplerangelenshapecat)argsrr r!r"x_origr(cond_eps uncond_eps cond_rest_half_eps half_rest ori_pos_norm max_new_norm new_pos_norm cfg_resultrrrrrrenorm_cfg_funcs8  **   **z*RenormCFG.execute..renorm_cfg_func)cloneset_model_sampler_cfg_functionr NodeOutput)rrrrr?mrr>rexecutes$  zRenormCFG.executeN)__name__ __module__ __qualname__ classmethodrrrBrDrrrrrs  rc@s:eZdZdddZdZeddZedejfdd Z d S) CLIPTextEncodeLumina2zYou are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.zYou are an assistant designed to generate high-quality images with the highest degree of image-text alignment based on textual prompts.)superior alignmenta]Lumina2 provide two types of system prompts:Superior: You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts. Alignment: You are an assistant designed to generate high-quality images with the highest degree of image-text alignment based on textual prompts.c Csbtjddgdddtjjdt|j|jdtjjdd d d d tj jd d dgtj j ddgdS)NrIz lumina promptzCLIP Text Encode for Lumina2 conditioningzEncodes a system prompt and a user prompt using a CLIP model into an embedding that can be used to guide the diffusion model towards generating specific images. system_prompt)optionstooltip user_promptTzThe text to be encoded.) multilinedynamic_promptsrOclipz*The CLIP model used for encoding the text.)rOzNA conditioning containing the embedded text used to guide the diffusion model.)rsearch_aliases display_namer descriptionrr) rrComborlist SYSTEM_PROMPTkeysSYSTEM_PROMPT_TIPStringClip ConditioningrrrrrrNs0 z#CLIPTextEncodeLumina2.define_schemarcCsB|durtd|j|}|d|}||}t||S)NzERROR: clip input is invalid: None If the clip is from a checkpoint loader node your checkpoint does not contain a valid clip or text encoder model.z ) RuntimeErrorrYtokenizerrBencode_from_tokens_scheduled)rrSrPrMprompttokensrrrrDls   zCLIPTextEncodeLumina2.executeN) rErFrGrYr[rHrrrBrDrrrrrIBs rIc@s(eZdZedeeejfddZdS)Lumina2Extensionrcs ttgSN)rIr)selfrrr get_node_listwszLumina2Extension.get_node_listN) rErFrGrrXtyper ComfyNodergrrrrrdvsrdrcstSre)rdrrrrcomfy_entrypointsrj) typing_extensionsrr*comfy_api.latestrrrirrIrdrjrrrrs ;4