bi~dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZmZddlZddlmZmZmZddlmZmZmZmZddlmZdd lm Z d d l!m"Z"d d l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)e)jTe+Z,ejZd Z.Gdde Z/GddZ0dZ1dZ2Gdde0Z3y)z%ConfigMixin base class and utilities.N) OrderedDict)Path)AnyDictOptionalTupleUnion) DDUFEntry create_repohf_hub_download)EntryNotFoundErrorRepositoryNotFoundErrorRevisionNotFoundErrorvalidate_hf_hub_args) HTTPError)Self) __version__)HUGGINGFACE_CO_RESOLVE_ENDPOINT DummyObject deprecateextract_commit_hashhttp_user_agentloggingzconfig\.(.*)\.jsoncJeZdZfdZdZdZdZdZfdZfdZ xZ S) FrozenDictc|t||i||jD]\}}t|||d|_y)NT)super__init__itemssetattr_FrozenDict__frozen)selfargskwargskeyvalue __class__s X/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/configuration_utils.pyrzFrozenDict.__init__9sC $)&)**, &JC D#u % & cHtd|jjd)Nz$You cannot use ``__delitem__`` on a instance. Exceptionr(__name__r#r$r%s r) __delitem__zFrozenDict.__delitem__As#>t~~?V?V>WWabccr*cHtd|jjd)Nz#You cannot use ``setdefault`` on a r,r-r0s r) setdefaultzFrozenDict.setdefaultDs#=dnn>U>U=VV`abbr*cHtd|jjd)NzYou cannot use ``pop`` on a r,r-r0s r)popzFrozenDict.popGs"6t~~7N7N6OzZ[[r*cHtd|jjd)NzYou cannot use ``update`` on a r,r-r0s r)updatezFrozenDict.updateJs#9$..:Q:Q9RR\]^^r*ct|dr/|jr#td|jjdt |||yN__frozenz$You cannot use ``__setattr__`` on a r,)hasattrr"r.r(r/r __setattr__r#namer'r(s r)r<zFrozenDict.__setattr__MC 4 $B4>>CZCZB[[efg g D%(r*ct|dr/|jr#td|jjdt |||yr9)r;r"r.r(r/r __setitem__r=s r)rAzFrozenDict.__setitem__Rr?r*) r/ __module__ __qualname__rr1r3r5r7r<rA __classcell__r(s@r)rr8s/dc\_) ))r*rc eZdZdZdZgZdZgZdZde de fdZ dde e e jfd efd Ze dd e eee e ffde eeeee e ffffd Zed Zee dde e e jfdeee e fee e fffdZedZedZe dde e e jfdeee effdZdZ e!dee e ffdZ"de fdZ#de e e jffdZ$ede dee effdZ%y) ConfigMixina Base class for all configuration classes. All configuration parameters are stored under `self.config`. Also provides the [`~ConfigMixin.from_config`] and [`~ConfigMixin.save_config`] methods for loading, downloading, and saving classes that inherit from [`ConfigMixin`]. Class attributes: - **config_name** (`str`) -- A filename under which the config should stored when calling [`~ConfigMixin.save_config`] (should be overridden by parent class). - **ignore_for_config** (`List[str]`) -- A list of attributes that should not be saved in the config (should be overridden by subclass). - **has_compatibles** (`bool`) -- Whether the class has compatible classes (should be overridden by subclass). - **_deprecated_kwargs** (`List[str]`) -- Keyword arguments that are deprecated. Note that the `init` function should only have a `kwargs` argument if at least one argument is deprecated (should be overridden by subclass). NFc 0|jtd|jd|jddt |ds|}n@t |j }i|j |}tjd|d|t||_y)NzMake sure that z' has defined a class name `config_name`r%_internal_dictzUpdating config from z to ) config_nameNotImplementedErrorr(r5r;dictrIloggerdebugr)r#r% internal_dict previous_dicts r)register_to_configzConfigMixin.register_to_configos    #%7GGn&op p  8T"t-."M !4!45M=t22=f=M LL0tM?S T(7r*r>returnc nd|jvxrt|jd|}||jv}|rX|sVd|dt|jd|dt|jd|d }t dd |d |j |St d t|jd |d )aWThe only reason we overwrite `getattr` here is to gracefully deprecate accessing config attributes directly. See https://github.com/huggingface/diffusers/pull/3129 This function is mostly copied from PyTorch's __getattr__ overwrite: https://pytorch.org/docs/stable/_modules/torch/nn/modules/module.html#Module rIzAccessing config attribute `z` directly via 'z1' object attribute is deprecated. Please access 'z' over 'z1's config object instead, e.g. 'scheduler.config.z'.zdirect config name access1.0.0F standard_warn'z' object has no attribute ')__dict__r;typer/rrIAttributeError)r#r> is_in_config is_attributedeprecation_messages r) __getattr__zConfigMixin.__getattr__s(4==8kWT]]ScEdfj=k t}},  $@FVW[\`WaWjWjVkl]^b]ccklpqulvllk@@qrvqwwy#z  17>. ) ?>2BBe!fg g NT2 WW\\.$:J:JK ,- -.@-ABC #ZZ(8$?NjjD1G ;6IJJw-EjjN,@,@,Mb,QRG!'D'QVW__G ;5I   -##   r*configc Dd|vr|jd}| tdt|ts[d}d|jvr |d|dz }nd|jvr |d|d |d z }t d d |d |j d|dd|\}}|j|fi|\}}}d|vr|jd|d<|jD]}||vs|j|||<|di|} d|vr|j|d<| jdi|i||}|r| |fS| S)a  Instantiate a Python class from a config dictionary. Parameters: config (`Dict[str, Any]`): A config dictionary from which the Python class is instantiated. Make sure to only load configuration files of compatible classes. return_unused_kwargs (`bool`, *optional*, defaults to `False`): Whether kwargs that are not consumed by the Python class should be returned or not. kwargs (remaining dictionary of keyword arguments, *optional*): Can be used to update the configuration object (after it is loaded) and initiate the Python class. `**kwargs` are passed directly to the underlying scheduler/model's `__init__` method and eventually overwrite the same named arguments in `config`. Returns: [`ModelMixin`] or [`SchedulerMixin`]: A model or scheduler object instantiated from a config dictionary. Examples: ```python >>> from diffusers import DDPMScheduler, DDIMScheduler, PNDMScheduler >>> # Download scheduler from huggingface.co and cache. >>> scheduler = DDPMScheduler.from_pretrained("google/ddpm-cifar10-32") >>> # Instantiate DDIM scheduler class with same config as DDPM >>> scheduler = DDIMScheduler.from_config(scheduler.config) >>> # Instantiate PNDM scheduler class with same config as DDPM >>> scheduler = PNDMScheduler.from_config(scheduler.config) ``` pretrained_model_name_or_pathzFPlease make sure to provide a config as the first positional argument.zJIt is deprecated to pass a pretrained model name or path to `from_config`. Schedulerz3If you were trying to load a scheduler, please use z.from_pretrained(...) instead. Otherwise, please make sure to pass a configuration dictionary instead. This functionality will be removed in v1.0.0.Modelz/If you were trying to load a model, please use z.load_config(...) followed by z.from_config(...) instead. Otherwise, please make sure to pass a configuration dictionary instead. This functionality will be removed in v1.0.0.zconfig-passed-as-pathrTFrUT)ryreturn_unused_kwargsdtype _class_name) r5 ValueError isinstancerLr/r load_configextract_init_dict_deprecated_kwargsrQ) clsrwr|r%r] init_dict unused_kwargs hidden_dictdeprecated_kwargmodels r) from_configzConfigMixin.from_configsN +f 4ZZ ?@F >ef f&$'"n cll*#I#O--# CLL(#EcUKuNN# -w8K[` a,S__w6hlwpvwNFF0E0E0Ef0WPV0W- =+ m #!.!2!27!;Ig !$ 6 6 R =0.;.?.?@P.Q *+ R  i  K '),K &   /;/9=8K8 =) )Lr*cPd|d}tdd|d|j|i|S)Nz8 The function get_config_dict is deprecated. Please use zE.load_config instead. This function will be removed in version v1.0.0get_config_dictrTFrU)rr)rr$r%r]s r)rzConfigMixin.get_config_dictsEGseL) )  #W.AQVWs///r*ryc |jdd}|jdd}|jdd}|jdd}|jdd} |jd d} |jd d} |jd d} |jd d} |jd d}|jdi}|jdd}i|ddi}t|}t|}|j t d|r!| t d|j ||}nit jj|r|}nFt jj|r|tt jjt jj|||jr,t jj|||j}nt jjt jj||jr+t jj||j}n>td|jd|d t||j||| | | ||| || } |j%||&}t'|}|s|s|S|f}|r||fz }|r||fz }|S#t$rt|dt$rt| d|dt$rt|d|jdt $r}td|d|d}~wt$r$tdt"d|d |jd!t$r td"|d#|d$|jd%wxYw#t(j*t,f$rtd'|d(wxYw))a Load a model or scheduler configuration. Parameters: pretrained_model_name_or_path (`str` or `os.PathLike`, *optional*): Can be either: - A string, the *model id* (for example `google/ddpm-celebahq-256`) of a pretrained model hosted on the Hub. - A path to a *directory* (for example `./my_model_directory`) containing model weights saved with [`~ConfigMixin.save_config`]. cache_dir (`Union[str, os.PathLike]`, *optional*): Path to a directory where a downloaded pretrained model configuration is cached if the standard cache is not used. force_download (`bool`, *optional*, defaults to `False`): Whether or not to force the (re-)download of the model weights and configuration files, overriding the cached versions if they exist. proxies (`Dict[str, str]`, *optional*): A dictionary of proxy servers to use by protocol or endpoint, for example, `{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}`. The proxies are used on each request. output_loading_info(`bool`, *optional*, defaults to `False`): Whether or not to also return a dictionary containing missing keys, unexpected keys and error messages. local_files_only (`bool`, *optional*, defaults to `False`): Whether to only load local model weights and configuration files or not. If set to `True`, the model won't be downloaded from the Hub. token (`str` or *bool*, *optional*): The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from `diffusers-cli login` (stored in `~/.huggingface`) is used. revision (`str`, *optional*, defaults to `"main"`): The specific model version to use. It can be a branch name, a tag name, a commit id, or any identifier allowed by Git. subfolder (`str`, *optional*, defaults to `""`): The subfolder location of a model file within a larger model repository on the Hub or locally. return_unused_kwargs (`bool`, *optional*, defaults to `False): Whether unused keyword arguments of the config are returned. return_commit_hash (`bool`, *optional*, defaults to `False): Whether the `commit_hash` of the loaded configuration are returned. Returns: `dict`: A dictionary of all the parameters stored in a JSON configuration file. cache_dirN local_dirlocal_dir_use_symlinksautoforce_downloadFproxiesrflocal_files_onlyrevisionmirrorri user_agent dduf_entries file_typerwz`self.config_name` is not defined. Note that one should not load a config from `ConfigMixin`. Please make sure to define `config_name` in a class inheriting from `ConfigMixin`zDDUF file only allow for 1 level of directory (e.g transformer/model1/model.safetentors is not allowed). Please check the DDUF structurezError no file named z found in directory .) filenamerrrrrfrrirrrz is not a local folder and is not a valid model identifier listed on 'https://huggingface.co/models' If this is a private repository, make sure to pass a token having permission to this repo with `token` or log in with `hf auth login`.z is not a valid git identifier (branch name, tag name or commit id) that exists for this model name. Check the model page at 'https://huggingface.co/z' for available revisions.z& does not appear to have a file named z:There was a specific connection error when trying to load z: zWe couldn't connect to 'zM' to load this model, couldn't find it in the cached files and it looks like z- is not the path to a directory containing a z file. Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/diffusers/installation#offline-mode'.zCan't load config for 'z'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'z2' is the correct path to a directory containing a z file)rz"It looks like the config file at 'z' is not a valid JSON file.)r5rstrrJr_get_config_file_from_ddufrjrkrlisdirroEnvironmentErrorr rrr rr_dict_from_json_filerjsonJSONDecodeErrorUnicodeDecodeError)rryr|return_commit_hashr%rrrrrrfrr_rirr config_fileerr config_dict commit_hashoutputss r)rzConfigMixin.load_config#s3jJJ{D1 JJ{D1 !',Df!M$4e<**Y- 7D)!::&8%@::j$/ JJx &JJ{D1 ZZ b1 7=zz.RV7W : :K: $Z0 (+,I(J% ?? "s  $ 6889VXdeK WW^^9 :7K WW]]8 9$ :IsW*!ggll+H)UXUdUde  -JCOO \] ggll+H#//Z &*3??*;;OPmOnnop3 -1 __'#1#%5)'%'+A  d r22;\2ZK-k:K%(: .  y G   ~ %Gm+ &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&9!KNAN/MAN)Octttj|jj j SN)setrLinspect signaturer parameterskeys) input_classs r)_get_init_keyszConfigMixin._get_init_keyss24))+*>*>?JJKPPRSSr*c <|jdg}|jDcic]\}}||vs |dk7s||}}}t|j}|j|}|j dd|vr|j dt |dr"|j D]}|j |t|jdkDr|t|jz }tjtjdd} |jr/|jD cgc]} t!| t"r| } } ng} t} | D]$} |j| } | j%| } &| |j|z } |jDcic] \}}|| vs ||}}}|j'd|j}t!|t(rd||jk7rUt | |rIt+| |}|j||z }|jDcic] \}}||vs ||}}}n1t!|t(s!t!|t,t.fs t1d|jDcic]\}}|j3d r||}}}|jDcic]\}}|d k7s ||}}}i}|D]P}||vr||vr|j'|||<||vr|j'|||<8||vs=|j'|||<Rt|dkDr3t4j7d |d |jd |j8dt|j;}t||z dkDrt4j=||z di||}|jDcic] \}}||vs ||}}}|||fScc}}wcc} wcc}}wcc}}wcc}}wcc}}wcc}}w)N_use_default_valuesr#r%_flax_internal_argsrrr~z\Make sure that the `_class_name` is of type string or list of string (for custom pipelines).rquantization_configzThe config attributes z were passed to z?, but are not expected and will be ignored. Please verify your z configuration file.zG was not found in config. Values will be initialized to default values.)getr rLrremover;rlenignore_for_configr importlib import_moduler/rrhas_compatibles_get_compatiblesrrunionr5rgetattrlisttupler startswithrMwarningrJrrq)rrr% used_defaultskv original_dict expected_keysargdiffusers_libraryccompatible_classesexpected_keys_comp_clsexpected_keys_c orig_cls_nameorig_clsunexpected_keys_from_origrr& passed_keysrhidden_config_dicts r)rzConfigMixin.extract_init_dicts>$(=rB (3(9(9(;u1q ?UZ[_tZtq!tu u[..01 **3/ V$ } $   * 3- ... *$$S) * s$$ % ))C0E0E,FFM&33HNN34G4JK   -0-A-A-C!f:VWYdKe!!f !f!# !$# SA!003O%;%A%A/%R " S"8#:L:LS:Q!Q(3(9(9(;_1qH^?^q!t_ _$ s||D }c *-)=90-@H(+(:(:8(D}(T %,7,=,=,?fDAq1LeCe1a4fKfM3/ =SWY^R_8`n  )4(9(9(;U11<P P PP 4P  PP P(P json_filerc|r||j}n't|dd5}|j}dddtjS#1swYxYw)Nrutf-8encoding) read_textopenreadrloads)rrrtextreaders r)rz ConfigMixin._dict_from_json_file5sW  *446Diw7 %6{{} %zz$ % %s AAcT|jjd|jS)N )r(r/to_json_stringr#s r)__repr__zConfigMixin.__repr__@s(..))*!D,?,?,A+BCCr*c|jS)z Returns the config of the class as a frozen dictionary Returns: `Dict[str, Any]`: Config of the class. )rIrs r)rwzConfigMixin.configCs"""r*ct|dr |jni}|jj|d<t|d<fdd|vrCt |j ts|j jn |j |d<|jDcic]\}}||}}}|jdd|jdd|jd d}tj|d d d zScc}}w)z Serializes the configuration instance to a JSON string. Returns: `str`: String containing all the attributes that make up the configuration instance in JSON format. rIr~_diffusers_versionc^t|tjr|j}|St|tr|j }|St |dr't|jr|j}|St|tr|Dcgc] }| }}|Scc}w)Nto_dict) rnpndarraytolistras_posixr;callablerr)r'rto_json_saveables r)rz4ConfigMixin.to_json_string..to_json_saveableYs%, L E4(( L  *x /F LE4(6;<)!,<<L=sB*r _ignore_filesNr_pre_quantization_dtypeT)indent sort_keys ) r;rIr(r/rrrrLrr r5rdumps)r#rrrrrs @r)rzConfigMixin.to_json_stringMs.5T;K-Ld))RT %)^^%<%< M",7 ()  !K /"+"A"A4H//779 44 - . ;F:K:K:MN$!Qq*1--N N.-t4 OO5t <zz+a4@4GGOsDjson_file_pathct|dd5}|j|jdddy#1swYyxYw)z Save the configuration instance's parameters 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)rwriter)r#rwriters r)rpzConfigMixin.to_json_filets<.# 8 0F LL,,. / 0 0 0s 8Ac|dk(r |jndj||jg}||vrtd|d|j|S)N/z#We did not manage to find the file z4 in the dduf file. We only have the following files )rJrorr)rryrrs r)rz&ConfigMixin._get_config_file_from_ddufs -2 OO8#//JK  l *5k]BvxDxIxIxKwLM r*)FNF)FFr)&r/rBrC__doc__rJrrrrQrrr^r rjPathLikeboolrv classmethodrrrrrrrr staticmethodrrrr rrpropertyrwrrprrr*r)rGrGXs  KO8"ZZZ$*%R[[0@*A*PT*XTYY:tCH~56Y tU4c3h/00 1YYv00# g',S"++-='>g tCH~tCH~- . ggRTT`<`prrrinits r) inner_initz®ister_to_config..inner_inits$)/ P1all3>Oq!tP P/5||~Stq!cARadSS$ ,89P9P8QR22  2B7 %%d+ 09):N:N:T:T:V0W  ,9D![\_`[`eiqwewD!))O  T:??#45 #IC"Jt  # #-"2"2"4 AwF?q ':;??1g..   s:??$%K(88 9A =04S9J5KcR]N^5^0_J, -9*9j9 +*+9j9 T(D(K(CQS  s.H H  H$H*H;HH%H ) functoolswraps)rrs` r)rQrQs)__T#)#)J r*ch|jtjfd}||_|S)NcBt|ts#td|jjdt |j }tj|}i}|D]t}|j|jvrt|jtjk(rd||j<Rt||j||j<vi||}d|vr|jdt!|D]\}} ||j} | || <t#t%|j't%|z dkDr1t)t%|j't%|z |d<t|ddi| |g|i|y)Nr r r}rrrQr)rrGr r(r/rLr dataclassesfieldsr>rrYr _MISSING_TYPErr5rrrrr) r#r$r%rrdefault_kwargsfieldrrrr> original_inits r)rz%flax_register_to_config..inits$ ,89P9P8QR22  6<<>* ##D) GEzzT555EMM"k&?&??-1uzz*-4T5::-Fuzz* G76+6 j NN7 # o #FAs!9>>D"Jt  # s:??$%K(88 9A =04S9J5KcR]N^5^0_J, -+*+9j9d,T,V,r*)rrr)rrr"s @r)flax_register_to_configr#s6LLM__]#&-$&-PCL Jr*cHeZdZdZeddeeeee ffffd Z xZ S)LegacyConfigMixinz A subclass of `ConfigMixin` to resolve class mapping from legacy classes (like `Transformer2DModel`) to more pipeline-specific classes (like `DiTTransformer2DModel`). rwc |ddlm}|||}||urtt|||fi|S|j||fi|S)Nr)_fetch_remapped_cls_from_config)models.model_loading_utilsr'rr%r)rrwr|r%r'remapped_classr(s r)rzLegacyConfigMixin.from_configsV P9E S *NGPdohno o->--f6JUfU Ur*r) r/rBrCrrr rrrrrrDrEs@r)r%r%s8  Vz4S>'A!B V Vr*r%)4rrrrrrrjre collectionsrpathlibrtypingrrrrr numpyrhuggingface_hubr r r huggingface_hub.utilsr rrrrequestsrtyping_extensionsrrrutilsrrrrrr get_loggerr/rMcompile_re_configuration_filerrGrQr#r%rr*r)r7s , #44CC "   H %#$9:))@ssl/d-`V Vr*