3jSSKJr SSKrSSKJrJr "SS\R 5r"SS\R 5r"SS \5r S \ 4S jr g) )overrideN)ComfyExtensioniocP\rSrSr\S5r\S\R4Sj5rSr g) RenormCFGc*[R"SS[RRS5[RRSSSSSS S 9[RRS S SSSS S 9/[RR 5/S 9$)Nrz model/patchmodel cfg_truncdgY@g{Gz?T)defaultminmaxstepadvanced renorm_cfgg?)node_idcategoryinputsoutputs)rSchemaModelInputFloatOutputclss :/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_lumina2.py define_schemaRenormCFG.define_schemasyy"w'{CSeRVaef|ScuSWbfg !  returnc^^^UUU4SjnTR5nURU5 [R"U5$)Nc >USnUSnUSnUSnUSnTRRRnUST:aUSS2SU24USS2SU24pUSS2US24USS2US24pXXx- --n U n [T5S:a[R R U[[S[UR555S S 9n U [T5-n[R R U [[S[U R555S S 9nX:aXU- -n O.USS2SU24USS2SU24pUSS2US24USS2US24pUn U n [R"X/SS 9nUU- $) 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_func*RenormCFG.execute..renorm_cfg_funcs 1M"#45Ol+JJ'H']F++55AAK{Y&'4Q  _'EWXZf[fZfWfGg*,Q _=qR]R^?_1%h6K(LL% $s*#(<<#;#;H"'aX^^1D(E"FPT$<$L$0% 2C#CL#(<<#;#;$%aX^^9L0M*NX\$<$L$3#+l/J#K'4Q  _'EWXZf[fZfWfGg*,Q _=qR]R^?_1#% H#8a@JJ& &r")cloneset_model_sampler_cfg_functionr NodeOutput)rr r rrEms ``` rexecuteRenormCFG.executes3" 'H KKM ((9}}Qr"N) __name__ __module__ __qualname____firstlineno__ classmethodr rrIrK__static_attributes__rMr"rrrs3    ' bmm' ' r"rc^\rSrSrSSS.rSr\S5r\S\R4Sj5r S r g ) CLIPTextEncodeLumina2BzYou 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.cz[R"SS/SSS[RRS[ UR R 55URS9[RRSS S S S 9[RRS S S9/[RRSS9/S9$)NrUz lumina promptzCLIP Text Encode (Lumina 2)zmodel/conditioning/luminazEncodes 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_promptsr\clipz*The CLIP model used for encoding the text.)r\zNA conditioning containing the embedded text used to guide the diffusion model.)rsearch_aliases display_namer descriptionrr) rrComborlist SYSTEM_PROMPTkeysSYSTEM_PROMPT_TIPStringClip Conditioningrrs rr #CLIPTextEncodeLumina2.define_schemaNsyy++,60l# !2!2!7!7!9:11  !"$(5    f.Z [ &&l'+  r"r#cUc [S5eURUnUSU3nURU5n[R"UR U55$)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 ) RuntimeErrorrftokenizerrIencode_from_tokens_scheduled)rr`r]rZprompttokenss rrKCLIPTextEncodeLumina2.executelse < yz z))-8 !?"2;-@v&}}T>>vFGGr"rMN) rNrOrPrQrfrhrRr rrIrKrSrMr"rrUrUBsTWOM C   :H"--HHr"rUcL\rSrSr\S\\\R4Sj5r Sr g)Lumina2Extensionvr#c"# [[/$7fN)rUr)selfs r get_node_listLumina2Extension.get_node_listws "   s rMN) rNrOrPrQrretyper ComfyNoderzrSrMr"rruruvs)  T$r||*<%=  r"rur#c# [5$7frx)rurMr"rcomfy_entrypointrs  s ) typing_extensionsrr1comfy_api.latestrrr}rrUrurrMr"rrsK& /8  8 v1HBLL1Hh ~  0r"