ij ddlZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z ddl mZmZddlmZmZmZmZmZddlmZmZmZdd lmZej8eZhd Zid d d ddddddddddddddddddddddddddddddddd d d!d ddd"d#d$d%d&d'd(d)d$d%d&d'd*d+dd,d-dd.d/d0d1d2d3d4d5d6d7d8d9d:dd1d;gdd:d?d:ddd@dAd3d,dBddCdDdEdFd3dGdddHdDddecoderdenoise controlnet vae_encoder text_encoder control_imagez Control Imageimageinput)labeltypedisplaycontrolnet_conditioning_scaleScalefloatg?)r r!defaultminmaxcontrol_guidance_endEndg?control_guidance_startStartgr Controlnetcustom_controlnet embeddingszText Embeddings)r r"r!Imagenegative_promptzNegative Promptstringtextarea)r r!r&r"promptPromptguidance_scalezGuidance Scalesliderg>@g?)r r!r"r&r'r(stepheightHeightinti@)r r!r&r'r: image_latentsz Image Latentslatentswidthstrength)FT)r r!r"onChangeLatentsnum_inference_stepsStepsd)r r!r"r&r'r(seedSeedrandomlStrengthg{Gz?)r r!r&r'r(r:Width ip_adapterz IP Adaptercustom_ip_adapter Schedulerdiffusers_auto_modelz Text Encodersdiffusers_auto_modelsUnetsignalguider)actiontarget)r r"r!onSignalGuider custom_guider)r r"r!rDVAEzControlnet Model) scheduler text_encodersunetrVvaeroutputImageszLatents Previewlatent)r/imagesr@rAlatents_previewcontrolnet_out)rr#r+r)r;rBr`controlnet_vae_encoder)inputs model_inputsoutputs block_names) r/rBr;rJrFr7r@rCrrO)r_rVr]rer)rrBr;r)r5r1rr^rrddecode)rrrrrT)frozenceZdZUeed<eed<eed<dZeeed<dZeed<dZ ee ed<dZ ee ed<dZ ee ed <dZ eed <dZeed <dZeeeefed <dZeed <dZeed<dZeed<deeeffdZy) MellonParamnamer r!Nr"r&r'r(r:optionsvalue fieldOptionsrDrY _map_to_inputreturnct|}|jDcic]\}}|jdr|||c}}Scc}}w)N_)ritems startswith)selfdatakvs w/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/modular_pipelines/mellon_node_utils.pyto_dictzMellonParam.to_dictKs<d|!%YAQ\\#5F1=1YYYsAAA)__name__ __module__ __qualname__str__annotations__r"rr&rr'r%r(r:rqrrrsrrDrYrtrr~roro:s I J I!GXc]!GSC%C% D(5/ GSE3-1L(4S>*1HcHcM3Zc3hZrroc eZdZUdZeeeefed<eeeefed<eeeefed<e eed<eed<dZ dZ e d eeeefd e ee eeffd e ee eefffd Zee dd eeej&fd ee eefe eefffdZddeeej&fdefdZdeeej&ffdZd efdZd e eeffdZede eefd dfdZededed dfdZy)MellonNodeConfigz A MellonNodeConfig is a base class to build Mellon nodes UI with modular diffusers. This is an experimental feature and is likely to change in the future. rhrirj blocks_names node_typezmellon_config.jsonct|jtr%|j|jt|_t|j tr%|j|j t |_t|jtr&|j|jt|_yyN) isinstancerhlist_resolve_params_listMELLON_INPUT_PARAMSriMELLON_MODEL_PARAMSrjMELLON_OUTPUT_PARAMSrzs r~ __post_init__zMellonNodeConfig.__post_init__cs dkk4 (33DKKATUDK d'' . $ 9 9$:K:KM` aD  dllD )44T\\CWXDL *rparams default_mapruc >dtttfdtttttffdt ttttfffd}i}|D]>}t jd||||\}}||vrtd|d|||<@|S)Nparamdefault_params_mapruc t|tr(||vrtd|d|||jfSt|tr%|j }|j d}||fStdt|d)NzUnknown param 'z/', please define a `MellonParam` object insteadrpzUnknown param type 'z9', please use a string or a `MellonParam` object instead)rr ValueErrorcopyrorpopr!)rr param_dict param_names r~_resolve_paramz=MellonNodeConfig._resolve_params_list.._resolve_paramos%% 22$ug=l%mnn07<<>>>E;/"]]_ '^^F3 !:-- *4;-7pqrz Resolving param: zDuplicate param '') r rrorrr loggerinfor)rrrresolvedprpcfgs r~rz%MellonNodeConfig._resolve_params_listks k)* @DS$sTWx.EX@Y  3S#X& '   !A KK,QC0 1&q+6ID#x #4TF!!<== HTN  ! rpretrained_model_name_or_pathc |jdd}|jdd}|jdd}|jdd}|jdd} |jd d} |jd d} |jd d} t|}|j td tj j |r|} ntj j|rtj j tj j||jr+tj j||j} n>7 87K WW]]8 9ww~~bggll+H#//Z[ ggll+H#//Z &*3??*;;OPmOnnop1 -1 __'#1#%5%'+A  ` rk39 %V{{} %**T*K-k:K%(: .  y G   ~ %Gq+ &456ii ) &j!00M/NNhj & &455[\_\k\k[llmn" &56c#@ &./N.OP>>[=\]//2.?@ZZ$ &-.K-LM--J,KL$$'OO#4E;  % % $$&89 r"%G }To#pq q rs>H#K1K (KAK I11AK KK)Lsave_directory push_to_hubc tjj|rtd|dtj|dtjj ||j }|j|tjd||r|jdd}|jdd}|jd d }|jd d}|jd |jtjjd } t| d||j} |jdd} |j|| |||| yy)a Save the Mellon node definition to a JSON file. Args: save_directory (`str` or `os.PathLike`): Directory where the configuration JSON file is saved (will be created if it does not exist). push_to_hub (`bool`, *optional*, defaults to `False`): Whether or not to push your model to the Hugging Face Hub after saving it. You can specify the repository you want to push to with `repo_id` (will default to the name of `save_directory` in your namespace). kwargs (`Dict[str, Any]`, *optional*): Additional keyword arguments passed along to the [`~utils.PushToHubMixin.push_to_hub`] method. zProvided path (z#) should be a directory, not a fileT)exist_okz Mellon node definition saved in commit_messageNprivate create_prFrrepo_id)rrr subfolder)rrrr)rrrAssertionErrormakedirsrr to_json_filerrrsplitsepr r_upload_folder) rzrrroutput_config_filerrrrrrs r~save_mellon_configz#MellonNodeConfig.save_mellon_config s7 77>>. ) ?>2BBe!fg g NT2 WW\\.$:J:JK ,- 67I6JKL #ZZ(8$?NjjD1G ;6IJJw-EjjN,@,@,Mb,QRG!'D'QVW__G ;5I   -##   rjson_file_pathct|dd5}|j|jdddy#1swYyxYw)z Save the Mellon schema dictionary to a JSON file. Args: json_file_path (`str` or `os.PathLike`): Path to the JSON file to save a configuration instance's parameters. wrrN)rwriteto_json_string)rzrwriters r~rzMellonNodeConfig.to_json_fileKs<.# 8 0F LL,,. / 0 0 0s 8AcX|j}tj|dddzS)z Serializes this instance to a JSON string of the Mellon schema dict. Args: Returns: `str`: String containing all the attributes that make up this configuration instance in JSON format. rT)indent sort_keys )to_mellon_dictrdumps)rz mellon_dicts r~rzMellonNodeConfig.to_json_stringVs*))+ zz+a4@4GGrci}|j|jxsi|j|jxsi|j|jxsi|j|j |dS)zReturn a JSON-serializable dict focusing on the Mellon schema fields only. params is a single flat dict composed as: {**inputs, **model_inputs, **outputs}. )rrr)updaterhrirjrr)rz merged_paramss r~rzMellonNodeConfig.to_mellon_dictbso  T[[.B/T..4"5T\\/R0 --#  rrc .|jdi}i}i}i}|jD]?\}}|jdddk(r|||<!|jdddvr|||<;|||<A|||||jdg|jd S) a<Create a config from a Mellon schema dict produced by to_mellon_dict(). Splits the flat params dict back into inputs/model_inputs/outputs using the known key spaces from MELLON_INPUT_PARAMS / MELLON_MODEL_PARAMS / MELLON_OUTPUT_PARAMS. Unknown keys are treated as inputs by default. rr"r3rar!)rRrSrrrhrirjrr)getrx)rr flat_paramsrhrirjrrs r~from_mellon_dictz!MellonNodeConfig.from_mellon_dictss"ooh3 !#') "$&1&7&7&9 0 "J ~~i,8&0 #+/``+5 Z(%/z"  0%$<!ook2   rblocksc |tvrtd|dt|jj }t|}|j dg}|j dg}|j dg}|j D]%}||vs|jt|||d'|jD]C} | j|vs|jt| j| jddE|||||| S) z Create an instance from a ModularPipeline object. If a preset exists in NODE_TYPE_PARAMS_MAP for the node_type, use it; otherwise fall back to deriving lists from the pipeline's expected inputs/components/outputs. z Node type z not supportedrhrirjr)rpr r!r"rRr) NODE_TYPE_PARAMS_MAPrr sub_blockskeysrrequired_inputsappendroexpected_componentsrp) rrrrdefault_node_config inputs_listmodel_inputs_list outputs_listrequired_input_namecomponent_specs r~ from_blockszMellonNodeConfig.from_blockss2 0 0z)NCD DF--2245 29=5H5L5LXWY5Z ;N;R;RSace;f6I6M6MiY[6\ #)#9#9  "+5""08KReov %88 N""*;;!((+00,113 '  * %   rN)FF)F)rrr__doc__rr rrorrrr staticmethodrrr classmethodrrPathLiker rboolrrrrrrrrrr~rrPs sK'( ))uS+-.// %[() **s)N&KYU3 +,-<@d3PS8nAT! "8# T',S"++-='>T tCH~tCH~- . TTl)sBKK7G1H)W[)V 05bkk1A+B 0 H H S#X " 4S> >P  <( !6( 3( K]( ( rrc`eZdZdZdZdedeeeffdZ dedefdZ deeeeefffdZ y ) ModularMellonNodeRegistryzKRegistry mapping (pipeline class, blocks_name) -> list of MellonNodeConfig.c i|_d|_y)NF) _registry _initializedrs r~__init__z"ModularMellonNodeRegistry.__init__s!r pipeline_cls node_paramscP|js t|||j|<yrr_initialize_registryr)rzrrs r~registerz"ModularMellonNodeRegistry.registers!   &'2|$rruch|js t||jj|dSr)rrrr)rzrs r~rzModularMellonNodeRegistry.gets*   &~~!!,55rcH|js t||jSrrrs r~get_allz!ModularMellonNodeRegistry.get_alls   &~~rN) rrrrrr!rrrrrrrrr~r r s]U"3T3SBR=R8S3 66)96 dD.>)>$??@rr  params_mapregistryc i}|jD]Z\}}t|jdg|jdg|jdg|jdg|}|||<\|j||y)zLRegister all node-type presets for a given pipeline class from a params map.rhrirjrkrN)rxrrr)rrr node_configsrspec node_configs r~_register_preset_node_typesr sL%++-. 4&88Hb)."5HHY+-4  #. Y. lL1rctdd|_ ddlm}ddlm}t ||| ddlm }ddl m }t |||y #t$r tdwxYw#t$r td wxYw) zDInitialize the registry and register all available pipeline configs.zInitializing registryTr)QwenImageModularPipeline)QwenImage_NODE_TYPES_PARAMS_MAPz+Failed to register QwenImageModularPipeline) StableDiffusionXLModularPipeline)SDXL_NODE_TYPES_PARAMS_MAPz3Failed to register StableDiffusionXLModularPipelineN) printrqwenimage.modular_pipeliner"qwenimage.node_utilsr#r  Exception$stable_diffusion_xl.modular_pipeliner$stable_diffusion_xl.node_utilsr%)rr"r#r$r%s r~rrs !" HGHI#$<>]_ghOZN#$DF`bjk GEFFG OMNNOsAA A A5)*rloggingr dataclassesrrtypingrrrrr r huggingface_hubr r huggingface_hub.utilsr rrrrutilsrrrmodular_pipeliner getLoggerrrSUPPORTED_NODE_TYPESrrrrrorr rr rrrr~r5s  *::8YX3   8 $[v v$ &v v* +v8#9vB"CvL MvV" Wvb  cvn!ov@  AvN %w/ |D OvZ[vd evt   uvD EvT  Uvb#cvv& !' &  -.b9 &  $&?$R#  !  # 5L     11!&     "{16    &      ''"      !z UVr $ZZZ* j ~j  j \ .2"3S#X#672C\2"O#<Or