/;ji xdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZmZmZddlZddlmZmZddlmZmZmZmZddlmZd d lmZd d lm Z m!Z!m"Z"m#Z#m$Z$m%Z%e%j&e'Z(ej)d Z*Gd de Z+GddZ,dZ-dZ.Gdde,Z/dS)z%ConfigMixin base class and utilities.N) OrderedDict) PosixPath)AnyDictTupleUnion) create_repohf_hub_download)EntryNotFoundErrorRepositoryNotFoundErrorRevisionNotFoundErrorvalidate_hf_hub_args) HTTPError) __version__)HUGGINGFACE_CO_RESOLVE_ENDPOINT DummyObject deprecateextract_commit_hashhttp_user_agentloggingzconfig\.(.*)\.jsoncJeZdZfdZdZdZdZdZfdZfdZ xZ S) FrozenDictctj|i||D]\}}t|||d|_dS)NT)super__init__itemssetattr_FrozenDict__frozen)selfargskwargskeyvalue __class__s W/root/voice-cloning/.venv/lib/python3.11/site-packages/diffusers/configuration_utils.pyrzFrozenDict.__init__8sZ$)&)))**,, & &JC D#u % % % % c<td|jjd)Nz$You cannot use ``__delitem__`` on a instance. Exceptionr%__name__r r!r"s r& __delitem__zFrozenDict.__delitem__@s!bt~?Vbbbcccr'c<td|jjd)Nz#You cannot use ``setdefault`` on a r)r*r-s r& setdefaultzFrozenDict.setdefaultCs!adn>Uaaabbbr'c<td|jjd)NzYou cannot use ``pop`` on a r)r*r-s r&popzFrozenDict.popFs!Zt~7NZZZ[[[r'c<td|jjd)NzYou cannot use ``update`` on a r)r*r-s r&updatezFrozenDict.updateIs!]$.:Q]]]^^^r'ct|dr$|jrtd|jjdt ||dSN__frozenz$You cannot use ``__setattr__`` on a r))hasattrrr+r%r,r __setattr__r namer$r%s r&r9zFrozenDict.__setattr__L_ 4 $ $ h hf4>CZfffgg g D%(((((r'ct|dr$|jrtd|jjdt ||dSr6)r8rr+r%r,r __setitem__r:s r&r>zFrozenDict.__setitem__Qr<r') r, __module__ __qualname__rr.r0r2r4r9r> __classcell__)r%s@r&rr7sdddccc\\\___))))) )))))))))r'rc eZdZdZdZgZdZgZdZde de fdZ dde e e jfd efd Zedd e eee e fffd Zed Zee dde e e jfdeee e fee e fffdZedZedZede e e jffdZdZedee e ffdZde fdZ de e e jffdZ!dS) 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 <|jtd|jd|ddt |ds|}n?t |j}i|j|}td|d|t||_dS)NzMake sure that z' has defined a class name `config_name`r"_internal_dictzUpdating config from z to ) config_nameNotImplementedErrorr%r2r8dictrEloggerdebugr)r r" internal_dict previous_dicts r&register_to_configzConfigMixin.register_to_configns   #%&o&o&o&opp p  8T"""t-.. U"MM !455M=t2=f=M LLSSSMSS T T T(77r'r;returnc `d|jvot|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 rEzAccessing 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__r8typer,rrEAttributeError)r r; is_in_config is_attributedeprecation_messages r& __getattr__zConfigMixin.__getattr__s$(4=8kWT]ScEdfj=k=k t},  -  -#z#z#zW[\`WaWaWj#z#z^b#z#zlpqulvlvl#z#zrv#z#z#z  17>. ) ) h !f>!f!f!fgg g NT2222 W\\.$:JKK ,--- B.@BBCCC  #ZZ(8$??NjjE22G ;66IJJw--EjjN,@,@,M,Mb,QRRG!'D'QVWWW_G   -#        r'configc *d|vr|d}|tdt|tsPd}d|jvr |d|dz }nd|jvr |d |d |d z }t d d |d|jd|dd|\}}|j|fi|\}}}d|vr|d|d<|jD]}||vr||||<|di|} d|vr |j|d<| j di|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_pathNzFPlease 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-pathrPFrQT)rtreturn_unused_kwargsdtype _class_name) r2 ValueError isinstancerHr,r load_configextract_init_dict_deprecated_kwargsrM) clsrrrwr"rY init_dict unused_kwargs hidden_dictdeprecated_kwargmodels r& from_configzConfigMixin.from_configsJ +f 4 4ZZ ?@@F >eff f&$'' x"n cl**#-#---## CL((#NcNNNNN# -w8K[` a a a a,S_w6hlwwpvwwNFF0E0Ef0W0WPV0W0W- =+ m # #!.!2!27!;!;Ig !$ 6 R R =00.;.?.?@P.Q.Q *+  i   K ' '),K &  //;///9=8K8  =) )Lr'cNd|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_dictrPFrQ)rr})rr!r"rYs r&rzConfigMixin.get_config_dictsP )s ) ) )  #W.AQVWWWWs////r'rtc |dd}|dd}|dd}|dd}|dd} |d d} |d d} |d d} |d d} |d d}|dd}|di}i|ddi}t|}t|}|jt dt j|r|}n t j|r|kt jt j |||jr(t j |||j}n}t jt j ||jr't j ||j}ntd|jd|d t||j||| | | | ||| || }n#t$rt|dt$rt| d|dt$rt|d|jdt$r}td|d|d}~wt$r$tdt d|d|jd t$rtd!|d"|d#|jd$wxYw ||}t%|}n-#t&jt*f$rtd%|d&wxYw|s|s|S|f}|r||fz }|r||fz }|S)'aN 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. resume_download: Deprecated and ignored. All downloads are now resumed by default when possible. Will be removed in v1 of Diffusers. 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_downloadFresume_downloadproxiesrblocal_files_onlyrevisionmirror subfolder user_agent file_typerrz`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`zError no file named z found in directory .) filenamerrrrrrbrrrrrz 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 `huggingface-cli 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 filez"It looks like the config file at 'z' is not a valid JSON file.)r2rstrrFr{rerfrgisdirrjEnvironmentErrorr r r r rr_dict_from_json_filerjsonJSONDecodeErrorUnicodeDecodeError)rrtrwreturn_commit_hashr"rrrrrrrbrr_rr config_fileerr config_dict commit_hashoutputss r&r}zConfigMixin.load_configs pJJ{D11 JJ{D11 !',Df!M!M$4e<< **%6==**Y-- 7D))!::&8%@@::j$// JJx & &JJ{D11 ZZ b11 : :K:: $Z00 (+,I(J(J% ? "s  7>>7 8 8C 7KK W]]8 9 9A $ :IsWW**$!gll+H)UXUdee  -JCO \ \]]  gll+H#/ZZ &p3?ppPmppp4 -1 _'#1#$3%5)'%'+A +   &4qqq )   &jj0Mjjj &   &4nn\_\knnn   &@5@@:=@@   &Z/NZZ>[ZZ/2ZZZ$   &;.K;;-J;;%(O;;;  r22;??K-k::KK$&89 r r r"#p #p#p#pqq q r% (:  .  ! y G  &  ~ %Gs&9!JA&M.LAM.2$N*Octttj|jjSN)setrHinspect signaturer parameterskeys) input_classs r&_get_init_keyszConfigMixin._get_init_keyss54)+*>??JKKPPRRSSSr'c |dgfd|D}t|}||}|dd|vr|dt |dr|jD]}||t|jdkr|t|jz }tj t dd}|jrd|D}ng}t|D],}||} | -||z fd |D}|d |j } t%| t&rd| |j krYt || rIt)|| } || |z fd |D}n@t%| t&s+t%| t*t,fst/d d |D}i|D][} | |vr| |vr|| || <| |vr|| | <?| |vr|| | <\t|dkr.t0d|d|j d|jdt} t|| z dkr t0|| z di||}fd|D}||fS)N_use_default_valuesc0i|]\}}|v |dk||S)rrz).0kv used_defaultss r& z1ConfigMixin.extract_init_dict..s5uuu1q ?U?UZ[_tZtZtq!ZtZtZtr'r r"_flax_internal_argsrrc<g|]}t|t|Srz)r|r)rcs r& z1ConfigMixin.extract_init_dict..s)!f!f!f:VWYdKeKe!f!!f!f!fr'c$i|] \}}|v || Srzrz)rrrexpected_keys_comp_clss r&rz1ConfigMixin.extract_init_dict..s*___1qH^?^?^q!?^?^?^r'ryc$i|] \}}|v || Srzrz)rrrunexpected_keys_from_origs r&rz1ConfigMixin.extract_init_dict..s*fffDAq1LeCeCe1aCeCeCer'z\Make sure that the `_class_name` is of type string or list of string (for custom pipelines).cDi|]\}}|d||Sr startswithrrrs r&rz1ConfigMixin.extract_init_dict..s0UUU11<.(s*[[[tq!QZHZHZaHZHZHZr')getrrHrremover8rlenignore_for_configr importlib import_moduler,rmhas_compatibles_get_compatiblesunionr2r|rgetattrlisttupler{rIwarningrFrrl)rrr" original_dict expected_keysargdiffusers_librarycompatible_classesrexpected_keys_c orig_cls_nameorig_clsr# passed_keysrhidden_config_dictrrrrs @@@@r&r~zConfigMixin.extract_init_dicts`$(=rBB uuuu (9(9(;(;uuu [..0011 **3// V$$$ } $ $   * * * 3- . . *. * *$$S)))) s$ % % ) ))C0E,F,FFM&3HNN34G4G4JKK   $!f!fS-A-A-C-C!f!f!f  !# !$# S SA!0033O%;%A%A/%R%R " "!7#:L:LS:Q:Q!Q____ (9(9(;(;___ $ s|DD }c * * --)=99.0-@@H(+(:(:8(D(D}(T %ffffK,=,=,?,?fffKKM3//  =SWY^R_8`8` n  VU (9(9(;(;UUU   6 6Cf}} !3!3#)::c?? C f}}!'C # ##!,!5!5 # {  a   NN999cl99?999   )..**++ }{* + +a / / KK ;.www    2;1&1 \[[[}/B/B/D/D[[[-);;;r' json_filect|dd5}|}dddn #1swxYwYtj|S)Nrutf-8encoding)openreadrloads)rrreadertexts r&rz ConfigMixin._dict_from_json_file,s )S7 3 3 3 !v;;==D ! ! ! ! ! ! ! ! ! ! ! ! ! ! !z$s 488cH|jjd|S)N )r%r,to_json_stringr s r&__repr__zConfigMixin.__repr__2s'.)CCD,?,?,A,ACCCr'c|jS)z Returns the config of the class as a frozen dictionary Returns: `Dict[str, Any]`: Config of the class. )rErs r&rrzConfigMixin.config5s ""r'c:t|dr|jni}|jj|d<t|d<dfd|D}|dd|ddtj|d d d zS) 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. rEry_diffusers_versionct|tjr|}n$t|trt |}|Sr)r|npndarraytolistrr)r$s r&to_json_saveablez4ConfigMixin.to_json_string..to_json_saveableKsG%,, # E9-- #E Lr'c.i|]\}}||Srzrz)rrrrs r&rz.ConfigMixin.to_json_string..Rs+NNN$!Qq**1--NNNr' _ignore_filesNrT)indent sort_keys ) r8rEr%r,rrr2rdumps)r rrs @r&rzConfigMixin.to_json_string?s.5T;K-L-LTd))RT %)^%< M",7 ()   ONNN+:K:K:M:MNNN ...-t444z+a4@@@4GGr'json_file_pathct|dd5}||ddddS#1swxYwYdS)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&rkzConfigMixin.to_json_fileYs.# 8 8 8 0F LL,,.. / / / 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0s(AA A )FNF)FF)"r,r?r@__doc__rFrrrrMrrrZrrePathLikeboolrq classmethodrrrrrrr} staticmethodrr~rrpropertyrrrrkrzr'r&rCrCWsr KO888"ZZZZZZ$((%R[0@*A(PT((((TWWz4S>'A!BWWW[Wr00[0# ff',S"+-='>f tCH~tCH~- . fff[fPTT\T]<]<[]<~ U3 3C-D   [ DDD#S#X###X#HHHHH4 05bk1A+B 0 0 0 0 0 0r'rCcFtjfd}|S)a Decorator to apply on the init of classes inheriting from [`ConfigMixin`] so that all the arguments are automatically sent to `self.register_for_config`. To ignore a specific argument accepted by the init but that shouldn't be registered in the config, use the `ignore_for_config` class variable Warning: Once decorated, all private arguments (beginning with an underscore) are trashed and not sent to the init! c d|D d|D}t|tstd|jjdt |dgi tj }fdt|j D}t|| D] \}}| |<  fd|Dtt t z dkrAt!t t z  d <i|  t |d d i  |g|Ri dS) NcDi|]\}}|d||Srrrs r&rz:register_to_config..inner_init..qs/PPP1all3>O>OPq!PPPr'cDi|]\}}|d||Srrrs r&rz:register_to_config..inner_init..rs/SSStq!cARARSaSSSr'&`@register_for_config` was applied to A init method, but this class does not inherit from `ConfigMixin`.rc@i|]\}\}}|dk|v||jS)r)default)rir;pignores r&rz:register_to_config..inner_init..}sD   ,9D![\_`[`[`eiqwewewD!)ewewewr'cTi|]$\}}|v |v |||%Srz)r)rrrr init_kwargs new_kwargss r&rz:register_to_config..inner_init..sJ   AwF??q ':':;??1g..':':':r'rrrMrz)rr|rC RuntimeErrorr%r,rrr enumeraterziprr4rrr) r r!r"config_init_kwargsrrrr;rrrinits @@@r& inner_initz®ister_to_config..inner_initns1QP PPP SSv||~~SSS$ ,, 29P222  2B77 %d++     09):N:T:T:V:V0W0W   T:??#4#455 # #IC"Jt         ","2"2"4"4       s:??$$%%K(8(88 9 9A = =04S9J9J5K5KcR]N^N^5^0_0_J, -9*9j9 +*++99j999 T(D(((K(((((r') functoolswraps)rrs` r&rMrMes<_T#)#)#)#)#)J r'cb|jtjfd}||_|S)Nclt|tstd|jjdt |}tj|}i}|D][}|j |j vrt|j tj kr d||j <>t||j ||j <\i||}d|vr|dt!|D]\}} ||j } | || <t#t%|t%|z dkrAt)t%|t%|z |d<t|ddi| |g|Ri|dS)Nr r rxrrrMrz)r|rCrr%r,rHr dataclassesfieldsr;rrUr _MISSING_TYPErr2rrrrr) r r!r"rr!default_kwargsfieldrrrr; original_inits r&rz%flax_register_to_config..inits$ ,, 29P222  6<<>>** #D)) G GEzT555EM""k&???-1uz**-4T5:-F-Fuz**76+6 j NN7 # # # oo # #FAs!9>D"Jt   s:??$$%%K(8(88 9 9A = =04S9J9J5K5KcR]N^N^5^0_0_J, -+*++99j999 d,T,,,V,,,,,r')rrr)rrr%s @r&flax_register_to_configr&sGLM_]##&-&-&-&-$#&-PCL Jr'cPeZdZdZeddeeeee fffdZ dS)LegacyConfigMixinz A subclass of `ConfigMixin` to resolve class mapping from legacy classes (like `Transformer2DModel`) to more pipeline-specific classes (like `DiTTransformer2DModel`). NFrrc Dddlm}|||}|j||fi|S)Nr)_fetch_remapped_cls_from_config)models.model_loading_utilsr*r)rrrrwr"r*remapped_classs r&rzLegacyConfigMixin.from_configsI POOOOO98EE)~)&2FQQ&QQQr'r) r,r?r@rrrrrrrrrzr'r&r(r(sd RRz4S>'A!BRRR[RRRr'r()0rr rrrrrere collectionsrpathlibrtypingrrrrnumpyrhuggingface_hubr r huggingface_hub.utilsr r r rrequestsrrutilsrrrrrr get_loggerr,rIcompile_re_configuration_filerrCrMr&r(rzr'r&r:sT ,+ ######************88888888   H % %#$9::))))))))@K0K0K0K0K0K0K0K0\///d---`RRRRR RRRRRr'