irlddlZddlZddlmZddlmZmZmZddlm Z m Z m Z m Z m Z mZmZddlZddlmZmZddlmZddlmZmZer ej2eZGd d eZeGd d ZeGd dZeGddZeGddZ dZ!dZ"ddZ#ddZ$ddZ%ddZ&ddZ' ddZ(y)N) OrderedDict) dataclassfieldfields)AnyDictListLiteralOptionalTypeUnion) ConfigMixin FrozenDict)_is_single_file_path_or_url)is_torch_availableloggingceZdZdZdZy)InsertableDictct|j}|Dcgc]\}}||k7s ||f}}}|j|||f|j|j ||Scc}}wN)listitemsinsertclearupdate)selfkeyvalueindexrkvs |/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/diffusers/modular_pipelines/modular_pipeline_utils.pyrzInsertableDict.insert#slTZZ\"%*6DAqQ#X!Q66  US%L)  E 7s A-A-c |syg}t|jD]\}\}}t|trd|jd|j d}n1d|j jd|j j d}|j|dt|d|dd d j|zd zS) NzInsertableDict()zz$>#?qAYAY@ZZ\] LLA3c$s)Bxj: ; <& U(;;fDDN)r,r+ __qualname__rr3r4r#rr"s  Er4rc"eZdZUdZdZeeed<dZee ed<dZ eeed<dZ ee ed<e dddi Zeeeeefed <e d ddi Zeeed <e dddi Zeeed <e dddi Zeeed<dZeded<e dddi Zeeeeefed<dZdZdZedededefdZedeefdZedefdZedede eeeffdZ!d deee e eeffdefdZ"defdZ#y)! ComponentSpeca=Specification for a pipeline component. A component can be created in two ways: 1. From scratch using __init__ with a config dict 2. using `from_pretrained` Attributes: name: Name of the component type_hint: Type of the component (e.g. UNet2DConditionModel) description: Optional description of the component config: Optional config dict for __init__ creation pretrained_model_name_or_path: Optional pretrained_model_name_or_path path for from_pretrained creation subfolder: Optional subfolder in pretrained_model_name_or_path variant: Optional variant in pretrained_model_name_or_path revision: Optional revision in pretrained_model_name_or_path default_creation_method: Preferred creation method - "from_config" or "from_pretrained" Nname type_hint descriptionconfigloadingT)defaultmetadatapretrained_model_name_or_path subfoldervariantrevisionfrom_pretrained) from_configrEdefault_creation_methodFrepocj|j}|%|jtj|d|yyy)Nr@)rHr@object __setattr__)r repo_values r# __post_init__zComponentSpec.__post_init__hs8YY  !d&H&H&P   t%Dj Q'Q !r4cZt|j|j|jfS)z=Make ComponentSpec hashable, using load_id as the hash value.)hashr9load_idrGrs r#__hash__zComponentSpec.__hash__ms"TYY d.J.JKLLr4ct|tsy|j|jk(xr4|j|jk(xr|j|jk(S)z8Compare ComponentSpec objects based on name and load_id.F)r)r8r9rPrG)rothers r#__eq__zComponentSpec.__eq__qsT%/ II # N  - N,,0M0MM r4 componentreturnc<t|dr|jdk7rd}nt|tjj r t dt|tr$t|dstjdd}n&t d|d|jjd |j}t|tr|dk(r |j}nd }t|dr+|jdk7r|j|j}ni}|d ||||d |S) aCreate a ComponentSpec from a Component. Currently supports: - Components created with `ComponentSpec.load()` method - Components that are ConfigMixin subclasses but not nn.Modules (e.g. schedulers, guiders) Args: name: Name of the component component: Component object to create spec from Returns: ComponentSpec object Raises: ValueError: If component is not supported (e.g. nn.Module without load_id, non-ConfigMixin) _diffusers_load_idnullrEzeCannot create ComponentSpec from a nn.Module that was not created with `ComponentSpec.load()` method.z\Component was not created using `ComponentSpec`, defaulting to `from_config` creation methodrFz!Cannot create ComponentSpec from (z). Currently ComponentSpec.from_component() only supports: - components created with `ComponentSpec.load()` method - components that are a subclass of ConfigMixin but not a nn.Module (e.g. guider, scheduler).N)r9r:r<rGr6)hasattrrYr)torchnnModule ValueErrorrloggerwarningr-r,r<decode_load_id)clsr9rVrGr:r< load_specs r#from_componentzComponentSpec.from_component{s2( 92 3 8T8TX^8^&7 #)UXX__5 {I{3y*>?NNv+8'!7vQy?R?R?[?[>\]uv '' i -2I]2Z%%FF 92 3 8T8TX^8^**9+G+GHII 6Sj nw  r4ct|Dcgc]+}|jjdds |j-c}Scc}w)ux Return the names of all loading‐related fields (i.e. those whose field.metadata["loading"] is True). r=F)rr?getr9)rdfs r#loading_fieldszComponentSpec.loading_fieldss1 !'s P1qzz~~i/OPPPs !AAc|jdk(ry|jDcgc]}t||}}|Dcgc]}|dn| }}djd|DScc}wcc}w)z Unique identifier for this spec's pretrained load, composed of pretrained_model_name_or_path|subfolder|variant|revision (no empty segments). rFrZ|c3&K|] }|s| ywrr6).0ps r# z(ComponentSpec.load_id..s.aA.s)rGrjgetattrr0)rr!partsros r#rPzComponentSpec.load_idsr  ' '= 8+/+>+>+@Aaq!AA5:;19!+;;xx.5...B;s A# A(rPc|j}tj|}|dk(r|S|jd}t |D]#\}}|t |ks|dk(rdn||||<%|S)a Decode a load_id string back into a dictionary of loading fields and values. Args: load_id: The load_id string to decode, format: "pretrained_model_name_or_path|subfolder|variant|revision" where None values are represented as "null" Returns: Dict mapping loading field names to their values. e.g. { "pretrained_model_name_or_path": "path/to/repo", "subfolder": "subfolder", "variant": "variant", "revision": "revision" } If a segment value is "null", it's replaced with None. Returns None if load_id is "null" (indicating component not created with `load` method). rZrlN)rjdictfromkeyssplitr(len)rdrPrjresultrrr1parts r#rczComponentSpec.decode_load_ids$++-~. f M c"!' MGAt3~&&48FND~a() M  r4c ^|jt|jts td|xs|jxsi}t |jt r|jj|fi|}ntj|jjj}i}|jD]\}}||vs |||<|jD]\}}||vs |||<|jdi|}d|_ t|dr|j|_|S)z/Create component using from_config with config.z?`type_hint` is required when using from_config creation method.rZr<r6)r:r)r*r`r< issubclassrrFinspect signature__init__ parametersrrYr\)rr<kwargsrVsignature_params init_kwargsr!r"s r#createzComponentSpec.creates >> !DNND)I^_ _,4;;," dnnk 22226DVDI&001H1HITT K  '1((%&KN '  '1((%&KN ''55I'- $ 9h '#**DKr4c |jDcic]}||vs||j|}}|jDcic]}||j|t||!}}|jdd}| t dt |}|r$|j t d|j|j . ddlm }|j|fi||}|j|_n:|rt|j d nt|j d } | |fi||}||_ |jD]\} } t|| | |j |_|Scc}wcc}w#t$r } t d|jd| d} ~ wwxYw#t$r } t d|jd | d} ~ wwxYw) z%Load component using from_pretrained.r@Nz`pretrained_model_name_or_path` info is required when using `load` method (you can directly set it in `pretrained_model_name_or_path` field of the ComponentSpec or pass it as an argument)zW`type_hint` is required when loading a single file model but is missing for component: r) AutoModelzUnable to load z without `type_hint`: from_single_filerEz using load method: )rjpoprhrqr`rr:r9 diffusersrrE Exceptionr-r@rsetattrrPrY) rrrpassed_loading_kwargs load_kwargsr@is_single_filerrVe load_methodr!r"s r#loadzComponentSpec.load sBFATATAV h#Z]agZgfjjo!5 h hZ^ZmZmZopSVs155c74;MNNp p(38WY](^% ( 0N 55RS dnn4ijnjsjsitu  >> ! Y/5I556SmWbmflm '00DN "(:;T^^->?   W'(E__X^_ .K*%%' DAq D!Q  '+|| $U!ip$ Y ?499+=STUSV!WXX Y W ?499+=QRSQT!UVV Ws@ E.E.$E3E8$ F$8 F!FF!$ G -GG r)$r,r+r5__doc__r9r str__annotations__r:r r;r<rrr@r r rBrCrDrGr rHrMrRrU classmethodrrfrjpropertyrPrrcrrr6r4r#r8r8Gs$D(3- $Ix~$!%K#%#'FHZ 'EJSWclnrbsEt!8E#tCy.,A#Bt$R9d:KLIx}L"49d:KLGXc]L#DIt;LMHhsmMIZW%EFZ-2$)UZI[,\D(5d3i( )\R M 9 #9 #9 #9 9 vQtCyQQ  / / / S T#x}2D-E  LXeJS#X,F&GH^a8--r4r8c:eZdZUdZeed<eed<dZeeed<y) ConfigSpecz5Specification for a pipeline configuration parameter.r9r>Nr;) r,r+r5rrrrr;r r6r4r#rr=s? I L!%K#%r4rcleZdZUdZdZeed<dZeed<dZ eed<dZ e ed<dZ eed <dZ eed <d Zy) InputParamz%Specification for an input parameter.Nr9r:r>FrequiredrAr; kwargs_typec^d|jd|jrdndd|jdS)N<: roptionalz , default=>)r9rr>rQs r#r3zInputParam.__repr__Vs3499+Rdmm LJW[WcWcVddeffr4)r,r+r5rr9rrr:rr>rboolr;rr3r6r4r#rrKsG/D#IsGSHdKKgr4rcLeZdZUdZeed<dZeed<dZeed<dZ eed<dZ y) OutputParamz&Specification for an output parameter.r9Nr:rAr;rcd|jdt|jdr|jjdSt |jdS)Nrrr,r)r9r\r:r,rrQs r#r3zOutputParam.__repr__csT {"PZ8[T^^44uuv w adeiesesatuuv w r4) r,r+r5rrrr:rr;rr3r6r4r#rrZs-0 IIsKK r4rc|Dcgc]}|js|}}|Dcgc]}|jr|}}djd|D}djd|D}|}|r |r|d|n|}|Scc}wcc}w)a Format input parameters into a string representation, with required params first followed by optional ones. Args: inputs: List of input parameters with 'required' and 'name' attributes, and 'default' for optional params Returns: str: Formatted string of input parameters Example: >>> inputs = [ ... InputParam(name="prompt", required=True), ... InputParam(name="image", required=True), ... InputParam(name="guidance_scale", required=False, default=7.5), ... InputParam(name="num_inference_steps", required=False, default=50) ... ] >>> format_inputs_short(inputs) 'prompt, image, guidance_scale=7.5, num_inference_steps=50' r&c34K|]}|jywr)r9rnparams r#rpz&format_inputs_short..|sEEUZZEsc3RK|]}|jd|j!yw)=N)r9r>rs r#rpz&format_inputs_short..}s#Z |1U]]O<Zs%')rr0)inputsrrequired_inputsoptional_inputs required_str optional_str inputs_strs r#format_inputs_shortris +1CENNuCOC*0GuGOG99E_EEL99Z/ZZLJ8D |2l^4, DGsA:A:A?A?cg}|D]u}|j|vr |jd|jd1|j|jd|jz}n |j}|j|w|Dchc]}|j}}g}g}|D]G} | j|vr|j| j-|j| jIg} |r#| jddj||r#| jddj||r#| jddj|| rdj| Sd Scc}w) aI Formats intermediate inputs and outputs of a block into a string representation. Args: intermediate_inputs: List of intermediate input parameters required_intermediate_inputs: List of required intermediate input names intermediate_outputs: List of intermediate output parameters Returns: str: Formatted string like: Intermediates: - inputs: Required(latents), dtype - modified: latents # variables that appear in both inputs and outputs - outputs: images # new outputs only z Required(r'z*_z - inputs: r&z - modified: z - outputs:  z (none))r9r.rr0) intermediate_inputsrequired_intermediate_inputsintermediate_outputs input_partsinpinp_name inputs_setmodified_partsnew_output_partsoutrxs r#format_intermediates_shortrs_"K") 883 3   388*A6 7xxCOO$?#//188   x ()'::s#((:J:N#. 88z !  ! !#(( +  # #CHH - . F tyy'=&>?@ (>)B(CDE  2B(C'DEF &499V 8L8%;sE,c|syd|z}d|dzz}d|dzz}g}d}d} |j||d|D]} | jtk7r|| jnd} | j| jd| jn | j} || d | d } t | d r/| j s#| d z } | j| d | jz } | dz } | jr5tjdd| j}| |||}| d||z } |j| dj|S)aFormat a list of InputParam or OutputParam objects into a readable string representation. Args: params: List of InputParam or OutputParam objects to format header: Header text to use (e.g. "Args" or "Returns") indent_level: Number of spaces to indent each parameter line (default: 4) max_line_length: Maximum length for each line before wrapping (default: 115) Returns: A formatted string representing all parameters rA c,t|dra|jturO|jDcgc]%}t|dr |jn t |'}}ddj |dSt|dr |jSt |Scc}w)N __origin__r,zUnion[r&])r\rr __args__r,rr0)r:ttypess r# get_type_strz#format_params..get_type_strs 9l + 0D0D0MOXOaOab!71j#9QZZs1vEbEbDIIe,-Q/ /%,Y %Cy!!WYWcs*Bcj|j}g}g}d}|D]c}t||rdndz}|r7||z|kDr/|jdj||g}t|}N|j|||z }e|r |jdj|d|j|S)zFWrap text while preserving markdown links and maintaining indentation.rrr)rvrwr.r0) textindent max_lengthwordslines current_linecurrent_lengthword word_lengths r# wrap_textz format_params..wrap_texts   .Dd)Lqa@K r;resubr0)paramsheader indent_levelmax_line_length base_indent param_indent desc_indentformatted_paramsrrrtype_strr9 param_strdesc wrapped_descs r# format_paramsrs|  $K,*+L)*KX )0{mF8156+49OOs4J<0PR+0::+=%BSBSB_E%%&'ejeoeo#nTF#hZq9  5* %>>^+ ==,>%--!AAIT    66={EL]L]^D$T;HL 2k]<.9 9I *++. ;;' ((r4ct|d||S)aFormat a list of InputParam objects into a readable string representation. Args: input_params: List of InputParam objects to format indent_level: Number of spaces to indent each parameter line (default: 4) max_line_length: Maximum length for each line before wrapping (default: 115) Returns: A formatted string representing all input parameters Inputsr) input_paramsrrs r#format_input_paramsrs x OOr4ct|d||S)aFormat a list of OutputParam objects into a readable string representation. Args: output_params: List of OutputParam objects to format indent_level: Number of spaces to indent each parameter line (default: 4) max_line_length: Maximum length for each line before wrapping (default: 115) Returns: A formatted string representing all output parameters Outputsr) output_paramsrrs r#format_output_paramsrs  < QQr4c|syd|z}d|dzz}g}|j|d|r|jdt|D]\}}t|jdr|jjnt |j} ||j d| d} |jr| d|jz } g} |jD]'} t|| } | | j| d | )| r| d d j| d z } |j| |s|t|d z ks|jd dj|S)aFormat a list of ComponentSpec objects into a readable string representation. Args: components: List of ComponentSpec objects to format indent_level: Number of spaces to indent each component line (default: 4) max_line_length: Maximum length for each line before wrapping (default: 115) add_empty_lines: Whether to add empty lines between components (default: True) Returns: A formatted string representing all components rArrz Components:r,rz`)rrz [r&rrr) r.r(r\r:r,rr9r;rjrqr0rw) componentsrradd_empty_linesrcomponent_indentformatted_componentsr1rV type_namecomponent_descloading_field_values field_name field_values r#format_componentsr"s  $KlQ./;-{ ;<##B'"*-, 9-4I4G4G,TI   ( (Z]^g^q^qZr --inn-=S 2N  9#8#8"9: :N "#224 KJ!)Z8K&$++zl!K=,IJ K 499-A#B"C1E EN##N3 q3z?Q#66 ' ' +3,6 99) **r4c|syd|z}d|dzz}g}|j|d|r|jdt|D]x\}}||jd|jd} |jr| d|jz } |j| |sV|t |dz ksh|jdzd j |S) aFormat a list of ConfigSpec objects into a readable string representation. Args: configs: List of ConfigSpec objects to format indent_level: Number of spaces to indent each config line (default: 4) max_line_length: Maximum length for each line before wrapping (default: 115) add_empty_lines: Whether to add empty lines between configs (default: True) Returns: A formatted string representing all configs rArrzConfigs:z (default: r'rrr)r.r(r9r>r;rwr0) configsrrrr config_indentformatted_configsr1r< config_descs r#format_configsrYs  $K.s D Dsrr)r)striprvr0rwrrrr) routputsr; class_nameexpected_componentsexpected_configsoutput desc_lines aligned_desccomponents_str configs_strs r#make_doc_stringrs0FF:,d++ &&(..t4 yy D DD ,''s#67!;*+>QO.6))C 01A5$%5AF +&& !&q99F fF "7;;F Mr4)Argsrs)rr)rrT)rANNN))r|r collectionsr dataclassesrrrtypingrrr r r r r r]configuration_utilsrrloaders.single_file_utilsrutilsrr get_loggerr,rarr8rrrrrrrrrrrr6r4r#rs #00BBB 9C/   H %E[EJ rr rj && &  g g  g       :/9dK)\ P R4+n#(R 5r4