+hgSSKrSSKrSSKJr SSKJrJrJr SSKJ r J r J r J r J r JrJr SSKrSSKJrJr SSKJrJr \"5(a\R."\5r"SS \5r\"S S 55r\"S S 55r\"SS55r\"SS55rSrSr SSjr!SSjr"SSjr#SSjr$SSjr%SSjr&g)N) OrderedDict) dataclassfieldfields)AnyDictListLiteralOptionalTypeUnion) ConfigMixin FrozenDict)is_torch_availableloggingc \rSrSrSrSrSrg)InsertableDict!c[UR55nUVVs/sHupVXQ:wdM XV4PM nnnURX1U45 UR5 UR U5 U$s snnfN)listitemsinsertclearupdate)selfkeyvalueindexrkvs l/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/modular_pipelines/modular_pipeline_utils.pyrInsertableDict.insert"sgTZZ\"%*6EDAQX!E6  U%L)  E 7s A.A.c U(dg/n[UR55Hunup4[U[5(aSURSUR S3nO1SUR RSUR R S3nURUS[U5SUS35 M S S RU5-S -$) NzInsertableDict()zz$>#?qAYAY@ZZ\] LLA3c$s)Bxj: ; 7& U(;;fDDN)r-r, __qualname____firstlineno__rr4__static_attributes__r7r6r#rr!s  Er6rc\rSrSr%SrSr\\\S'Sr \\ \S'Sr \\\S'Sr \\ \S'\"SSS 0S 9r\\\\\4\S '\"S SS 0S 9r\\\S '\"SSS 0S 9r\\\S'\"SSS 0S 9r\\\S'Sr\S\S'SrSr\S\S\S\4Sj5r\S\\4Sj5r\S\4Sj5r\S\S\\\\44Sj5r SS\\\ \\\44S\4Sjjr!S\4Sjr"Sr#g) ComponentSpecFaSpecification 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 repo: Optional repo path for from_pretrained creation subfolder: Optional subfolder in repo variant: Optional variant in repo revision: Optional revision in repo default_creation_method: Preferred creation method - "from_config" or "from_pretrained" Nname type_hint descriptionconfigloadingT)defaultmetadatarepo subfoldervariantrevisionfrom_pretrained) from_configrJdefault_creation_methodcZ[URURUR45$)z=Make ComponentSpec hashable, using load_id as the hash value.)hashr>load_idrLrs r#__hash__ComponentSpec.__hash__es"TYY d.J.JKLLr6c[U[5(dgURUR:H=(a9 URUR:H=(a URUR:H$)z8Compare ComponentSpec objects based on name and load_id.F)r*r<r>rOrL)rothers r#__eq__ComponentSpec.__eq__isW%// II # N  - N,,0M0MM r6 componentreturncz[US5(aURS:waSnO[U[RR 5(a [ S5e[U[5(a)[US5(d[RS5 SnO&[ SUSURRS 35eURn[U[5(aUS:Xa URnOS n[US5(a,URS:waURUR5nO0nU"S XXSS .UD6$) 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_idnullrJzeCannot 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 methodrKz!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)r>r?rArLr7)hasattrrZr*torchnnModule ValueErrorrloggerwarningr.r-rAdecode_load_id)clsr>rWrLr?rA load_specs r#from_componentComponentSpec.from_componentss?( 92 3 3 8T8TX^8^&7 #)UXX__55 {I{33y*>??NNv+8'!7vQy?R?R?[?[>\]uv '' i - -2I]2Z%%FF 92 3 3 8T8TX^8^**9+G+GHII 6 nw  r6c[U5Vs/sH1oRRSS5(dM%URPM3 sn$s snf)uh Return the names of all loading‐related fields (i.e. those whose field.metadata["loading"] is True). rBF)rrDgetr>)refs r#loading_fieldsComponentSpec.loading_fieldss5 !'s P 1zz~~i/O PPPs $A A cURS:XagUR5Vs/sHn[X5PM nnUVs/sH o3cSOUPM nnSRSU55$s snfs snf)zu Unique identifier for this spec's pretrained load, composed of repo|subfolder|variant|revision (no empty segments). rKr[|c36# UHo(dM Uv M g7frr7).0ps r# (ComponentSpec.load_id..s.5aA5s  )rLrlgetattrr1)rr!partsrrs r#rOComponentSpec.load_idsr  ' '= 8+/+>+>+@A+@a!+@A5:;U9!+U;xx.5...B;s A)A.rOcUR5nUVs0sHo3S_M nnUS:XaU$URS5n[U5H%upgU[U5:dMUS:XaSOUXBU'M' U$s snf)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: "repo|subfolder|variant|revision" where None values are represented as "null" Returns: Dict mapping loading field names to their values. e.g. { "repo": "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). Nr[ro)rlsplitr)len)rerOrlrkresultrvr2parts r#rdComponentSpec.decode_load_ids"++-#12>aT'>2 f M c"!'GA3~&&48FNDa()(  3s A2c URb[UR[5(d [S5eU=(d UR=(d 0n[ UR[ 5(aURR"U40UD6nO[R"URR5Rn0nUR5HupgXd;dM XuU'M UR5HupgXd;dM XuU'M UR"S0UD6nSUl [US5(aURUlU$)z/Create component using from_config with config.z?`type_hint` is required when using from_config creation method.r[rAr7)r?r*r+rarA issubclassrrKinspect signature__init__ parametersrrZr])rrAkwargsrWsignature_params init_kwargsr!r"s r#createComponentSpec.creates >> !DNND)I)I^_ _,4;;," dnnk 2 2226DVDI&001H1HITT K (%&N' (%&N'55I'- $ 9h ' '#**DKr6c UR5Vs0sHo"U;dM X!RU5_M nnUR5Vs0sHo"URU[X55_M nnURSS5nUc [ S5eUR c.SSKJn UR"U40UDUD6nURUlO UR R"U40UDUD6nXPl UR5Hup[X U 5 M URUlU$s snfs snf![a n[ SURSU35eSnAff=f![a n[ SURSU35eSnAff=f) z%Load component using from_pretrained.rENz`repo` info is required when using `load` method (you can directly set it in `repo` field of the ComponentSpec or pass it as an argument)r) AutoModelzUnable to load z without `type_hint`: z using load method: )rlpoprjrurar? diffusersrrJ Exceptionr>r.rErsetattrrOrZ) rrrpassed_loading_kwargs load_kwargsrErrWer!r"s r#loadComponentSpec.loadsBFATATAV hAV#agZg!5jjo!5AV hZ^ZmZmZopZoSV155c74;MNNZo pvt, <\  >> ! Y/%55dTkTVT '00DN W NN::4Y;YRXY  %%'DA DQ ('+|| $=!ip Y ?499+=STUSV!WXX Y W ?499+=QRSQT!UVV Ws@ D)D) %D.D3 E 3 E=EE F *FF )rArEr?r)$r-r,r8r9__doc__r>r str__annotations__r?r r@rArrrEr r rGrHrIrLr rQrU classmethodrrgrlpropertyrOrrdrrr:r7r6r#r<r<Fs$D(3- $Ix~$!%K#%#'FHZ ',1$)UYIZ,[D(5d3i( )[$R9d:KLIx}L"49d:KLGXc]L#DIt;LMHhsmMIZW%EFZM 9 #9 #9 #9 9 vQtCyQQ  / / /ST#x}2D-EJXeJS#X,F&GH^a8""r6r<cB\rSrSr%Sr\\S'\\S'Sr\ \\S'Sr g) ConfigSpeci)z5Specification for a pipeline configuration parameter.r>rCNr@r7) r-r,r8r9rrrrr@r r:r7r6r#rr)s? I L!%K#%r6rct\rSrSr%SrSr\\S'Sr\ \S'Sr \ \S'Sr \ \S'S r \\S 'Sr\\S 'S rS rg) InputParami7z%Specification for an input parameter.Nr>r?rCFrequiredrFr@ kwargs_typechSURSUR(aSOSSURS3$)N<: roptionalz , default=>)r>rrCrPs r#r4InputParam.__repr__Bs3499+Rdmm LJW[WcWcVddeffr6r7)r-r,r8r9rr>rrr?rrCrboolr@rr4r:r7r6r#rr7sG/D#IsGSHdKKgr6rcT\rSrSr%Sr\\S'Sr\\S'Sr \\S'Sr \\S'S r S r g) OutputParamiFz&Specification for an output parameter.r>Nr?rFr@rcSURS[URS5(aURRO[ UR5S3$)Nrrr-r)r>r]r?r-rrPs r#r4OutputParam.__repr__OsI {"PZ8[8[T^^44adeiesesatuuv w r6r7) r-r,r8r9rrrr?rr@rr4r:r7r6r#rrFs-0 IIsKK r6rc<UVs/sHoR(dMUPM nnUVs/sHoR(aMUPM nnSRSU55nSRSU55nUnU(aU(aUSU3OUnU$s snfs snf)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'c38# UHoRv M g7fr)r>rqparams r#rs&format_inputs_short..hsE_EZZ_sc3V# UHoRSUR3v M! g7f)=N)r>rCrs r#rsris!Z/ |1U]]Or/rr1) intermediate_inputsrequired_intermediate_inputsintermediate_outputs input_partsinpinp_name inputs_setmodified_partsnew_output_partsoutr{s r#format_intermediates_shortrrsV"K" 883 3   388*A6 7xxCOO$?#//188   x (#'::&9s((&9J:N# 88z !  ! !#(( +  # #CHH - $ F tyy'=&>?@ (>)B(CDE  2B(C'DEF &499V 8L8%;sFcU(dgSU-nSUS--nSUS--n/nSnSn URUUS35 UGH n U R[:waU"U R5OSn U RcU RbSU R3O U Rn UU S U S 3n [ U S 5(a5U R (d$U S - n U RbU S U R3- n U S- n U R(a5[R"SSU R5nU "XU5nU SUU3- n URU 5 GM SRU5$)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 rF cR[US5(ajUR[LaWURVs/sH+n[US5(a URO [ U5PM- nnSSR U5S3$[US5(a UR$[ U5$s snf)N __origin__r-zUnion[r'])r]rr __args__r-rr1)r?ttypess r# get_type_str#format_params..get_type_strs 9l + + 0D0D0MOXOaOabOa!71j#9#9QZZs1vEOaEbDIIe,-Q/ /%,Y %C%Cy!!WYWcs2B$cUR5n/n/nSnUHnn[U5U(aSOS-nU(a8Xh-U:a0URSRU55 U/n[U5nMYURU5 Xh- nMp U(a URSRU55 SU3RU5$)zFWrap text while preserving markdown links and maintaining indentation.rrr)ryrzr/r1) textindent max_lengthwordslines current_linecurrent_lengthword word_lengths r# wrap_text format_params..wrap_texts  Dd)Lqa@K rr]rrCr@resubr1)paramsheader indent_levelmax_line_length base_indent param_indent desc_indentformatted_paramsrrrtype_strr> param_strdesc wrapped_descs r# format_paramsrsx  $K,*+L)*KX )0{mF815649OOs4J<0PR+0::+=%BSBSB_E%%&'ejeoeo#nTF#hZq9  5* % %>>^+ ==,>%--!AAIT    66={EL]L]^D$THL 2k]<.9 9I *+. ;;' ((r6c[USX5$)akFormat 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 OOr6c[USX5$)aoFormat 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  < QQr6c U(dgSU-nSUS--n/nURUS35 U(aURS5 [U5GH!upx[URS5(aURRO[ UR5n UUR SU S3n UR(aU SUR3- n /n UR5H)n [X5n U cMU RU S U 35 M+ U (aU S S RU 5S 3- n URU 5 U(dMU[U5S - :dGMURS5 GM$ SRU5$)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 rFrrz Components:r-rz`)rrz [r'rrr) r/r)r]r?r-rr>r@rlrur1rz) componentsrradd_empty_linesrcomponent_indentformatted_componentsr2rW type_namecomponent_descloading_field_values field_name field_values r#format_componentsrs  $KlQ./;-{ ;<##B'"*- -4I4G4G,T,TI   ( (Z]^g^q^qZr --inn-=S 2N  9#8#8"9: :N "#224J!)8K&$++zl!K=,IJ5 499-A#B"C1E EN##N3 ?q3z?Q#66 ' ' +3.6 99) **r6cU(dgSU-nSUS--n/nURUS35 U(aURS5 [U5HupxUURSURS3n UR(aU SUR3- n URU 5 U(dM`U[ U5S- :dMtURS5 M S R U5$) 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 rFrrzConfigs:z (default: r(rrr)r/r)r>rCr@rzr1) configsrrrr config_indentformatted_configsr2rA config_descs r#format_configsr Es  $KUR5RS5nSRSU55nXhS-- nU(a [U5S:a[ USS9n XiS-- nU(a [U5S:a[ USS9n XjS-- nU[ USS9- nUS- nU[USS9- nU$) a Generates a formatted documentation string describing the pipeline block's parameters and structure. Args: inputs: List of input parameters intermediate_inputs: List of intermediate input parameters outputs: List of output parameters description (str, *optional*): Description of the block class_name (str, *optional*): Name of the class to include in the documentation expected_components (List[ComponentSpec], *optional*): List of expected components expected_configs (List[ConfigSpec], *optional*): List of expected configurations Returns: str: A formatted string containing information about components, configs, call parameters, intermediate inputs/outputs, and final outputs. rFzclass rrc3,# UH nSU-v M g7f)z Nr7)rqlines r#rs"make_doc_string..s Dsrr)r)stripryr1rzrr rr) routputsr@ class_nameexpected_componentsexpected_configsoutput desc_lines aligned_desccomponents_str configs_strs r#make_doc_stringrks0FF:,d++ &&(..t4 yy D DD ''s#67!;*+>QO6))C 01A5$%5AF && !&q99F fF "7;;F Mr6)Argsrs)rr)rrT)rFNNN)'rr collectionsr dataclassesrrrtypingrrr r r r r r^configuration_utilsrrutilsrr get_loggerr-rbrr<rrrrrrrrrr rr7r6r#r"s #00BBB 9/   H %E[EJ __ _D && &  g g  g       :/9dK)\ P R4+n#(R 5r6