+jddlmZddlZddlmZmZGddejZGddejZGdd eZ d e fd Z dS) )overrideN)ComfyExtensioniocJeZdZedZedejfdZdS) RenormCFGc8tjddtjdtjdddddd tjd d dddd gtjg S)Nrzadvanced/modelmodel cfg_truncdgY@g{Gz?T)defaultminmaxstepadvanced renorm_cfgg?)node_idcategoryinputsoutputs)rSchemaModelInputFloatOutputclss :/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_lumina2.py define_schemazRenormCFG.define_schemasy%w''{CSeRVaeff|ScuSWbfgg !!    returncfd}}||tj|S)Nc p|d}|d}|d}|d}|d}jjj}|dkr|ddd|f|ddd|f}}|dd|df|dd|df} } ||||z zz} | } tdkrtj|ttdt|j d } | tz}tj| ttdt| j d }||kr| ||z z} n<|ddd|f|ddd|f}}|dd|df|dd|df} } |} | } t j | | gd }||z S) N cond_denoiseduncond_denoised cond_scaletimestepinputrr T)dimkeepdim)r*) r diffusion_model in_channelsfloattorchlinalg vector_normtuplerangelenshapecat)argsr$r%r&r'x_origr-cond_eps uncond_eps cond_rest_half_eps half_rest ori_pos_norm max_new_norm new_pos_norm cfg_resultr r rs rrenorm_cfg_funcz*RenormCFG.execute..renorm_cfg_funcs 1M"#45Ol+JJ'H']F+5AK{Y&&'4QQQ  _'EWXWXWXZf[fZfWfGg*,QQQ _=qqqR]R^R^?_1 % h6K(LL% $$s**#(<#;#;H"'aX^1D1D(E(E"F"FPT$<$$L$0% 2C2C#CL#(<#;#;$%aX^9L9L0M0M*N*NX\$<$$L$|33#+|l/J#K'4QQQ  _'EWXWXWXZf[fZfWfGg*,QQQ _=qqqR]R^R^?_1 #% Hi#8a@@@JJ& &r )cloneset_model_sampler_cfg_functionr NodeOutput)rr r rrCms ``` rexecutezRenormCFG.executes\" '" '" '" '" '" '" 'H KKMM ((999}Qr N)__name__ __module__ __qualname__ classmethodrrrFrHr rrrsV   [  ' bm' ' ' [' ' ' r rcXeZdZdddZdZedZedejfdZ dS) 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.ctjddgdddtjdt |j|jtjdd d d tj d d gtj dgS)NrOz lumina promptzCLIP Text Encode for Lumina2zmodel/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_promptsrUclipz*The CLIP model used for encoding the text.)rUzNA conditioning containing the embedded text used to guide the diffusion model.)rsearch_aliases display_namer descriptionrr) rrComborlist SYSTEM_PROMPTkeysSYSTEM_PROMPT_TIPStringClip Conditioningrrs rrz#CLIPTextEncodeLumina2.define_schemaNsy++,7)l# !2!7!7!9!9::1  !"$(5    f.Z [[ &&l'+    r r!c|td|j|}|d|}||}tj||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 ) RuntimeErrorr_tokenizerrFencode_from_tokens_scheduled)rrYrVrSprompttokenss rrHzCLIPTextEncodeLumina2.executelsq < yzz z)-8 !@@;@@v&&}T>>vFFGGGr N) rIrJrKr_rarLrrrFrHrMr rrOrOBsWOM C   [ :H"-HHH[HHHr rOcLeZdZedeeejfdZdS)Lumina2Extensionr!c"KttgSN)rOr)selfs r get_node_listzLumina2Extension.get_node_listws "   r N) rIrJrKrr^typer ComfyNoderprMr rrlrlvsE  T$r|*<%=   X   r rlr!c"KtSrn)rlrMr rcomfy_entrypointrts   r ) typing_extensionsrr/comfy_api.latestrrrrrrOrlrtrMr rrws&&&&&& ////////8 8 8 8 8  8 8 8 v1H1H1H1H1HBL1H1H1Hh     ~    0r