bi2ZdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZmZmZddlmZddlmZmZddlmZmZdd lmZd d lmZd d lmZmZm Z e jBe"Z#dZ$e%ejLddZ'e jPZ)dZ*dZ+dee,ejZffdZ.dZ/dZ0dZ1dZ2dZ3d'dZ4dZ5e d(dee,ejZfde,deee,ejZfde6d eee,e,fd!eee6e,fd"ee,d#e6fd$Z7e d)dee,ejZfde,d%ee,deee,ejZfde6d eee,e,fd!eee6e,fd"ee,d#e6fd&Z8y)*z3Utilities to dynamically load objects from the Hub.N)Path) ModuleType)DictOptionalUnion)request)hf_hub_download model_info)RevisionNotFoundErrorvalidate_hf_hub_args)version) __version__)DIFFUSERS_DYNAMIC_MODULE_NAMEHF_MODULES_CACHEloggingz$diffusers/community-pipelines-mirrorDIFFUSERS_TIMEOUT_REMOTE_CODEcd}tjtj|j dj }t |dS)Nz$https://pypi.org/pypi/diffusers/jsonreleasesc,tj|S)N)r Version)xs `/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/utils/dynamic_modules_utils.pyz(get_diffusers_versions..2s'//!*<)key)jsonloadsrurlopenreadkeyssorted)urlrs rget_diffusers_versionsr&/sC 0Czz'//#.3356zBGGIH ( < ==rcttjvrytjjtt j tdt tdz }|js|jyy)z_ Creates the cache directory for modules with an init, and adds it to the Python path. NTexist_ok __init__.py) rsyspathappendosmakedirsrexiststouch) init_paths rinit_hf_modulesr35s^ 388#HHOO$%KK 40%&6I     rnamecttt|z }|jj st |jt j|d|dz }|j s|jyy)zF Creates a dynamic module in the cache directory for modules. Tr(r*N) r3rrparentr0create_dynamic_moduler.r/r1)r4dynamic_module_pathr2s rr7r7Dsp/047  % % , , .1889KK#d3#m3I     rc.t|dd5}|j}dddtjdtj}|tjd|tjz }t t |S#1swYlxYw)z Get the list of modules that are relatively imported in a module file. Args: module_file (`str` or `os.PathLike`): The module file to inspect. rutf-8encodingNz^\s*import\s+\.(\S+)\s*$flagsz^\s*from\s+\.(\S+)\s+import)openr"refindall MULTILINElistset) module_filefcontentrelative_importss rget_relative_importsrJSs} k3 1Q&&(zz"=wbll[ #A7RTR^R^__ $% &&s B  Bcld}|g}g}|sg}|D]}|jt|t|j}|Dcgc]}t ||z }}|Dcgc] }||vs| }}|Dcgc]}|d }}t |dk(}|j||s|Scc}wcc}wcc}w)a) Get the list of all files that are needed for a given module. Note that this function recurses through the relative imports (if a imports b and b imports c, it will return module files for b and c). Args: module_file (`str` or `os.PathLike`): The module file to inspect. F.pyr)extendrJrr6strlen) rF no_changefiles_to_checkall_relative_imports new_importsrG module_pathmnew_import_filess rget_relative_import_filesrWesI!]N  8A   3A6 7 8;'.. :EFQC a0FF'7Y!1DX;XAYY-=>QCs)>>()Q. ##N3 GY>sB'! B,+B,5 B1ct|dd5}|j}dddtjdtj}|tjd|tjz }|Dcgc](}|j dr|j dd *}}tt|}g}|D]} tj|t|d kDr0td d j|d d j|dt|S#1swYxYwcc}w#t$r|j|YwxYw)zi Check if the current Python environment contains all the libraries that are imported in a file. r:r;r<Nz^\s*import\s+(\S+)\s*$r>z^\s*from\s+(\S+)\s+import.rz\This modeling file requires the following packages that were not found in your environment: , z. Run `pip install  `)r@r"rArBrC startswithsplitrDrE importlib import_module ImportErrorr-rOjoinrJ)filenamerGrHimportsimpmissing_packagess r check_importsrgsF hg .!&&(jj2G2<<PG rzz6r||TTG,3OS3>>#;Nsyy~a OGO3w< G) )  # #C ()  q  jyy)*++>sxxHX?Y>ZZ[ ]   ))3P )  # #C ( )s)D&;D3D3D8&D08EEctd)NzLoading this model requires you to execute custom code contained in the model repository on your local machine. Please set the option `trust_remote_code=True` to permit loading of this model.) ValueError)signumframes r_raise_timeout_errorrls  c rc||rtdkDrd} tjtjt}tjt|>t d|d|d}|j dvrd}n|j dvrd}|>tjd |Htjtj|tjdn|r tdd|r|std |d |S#t$rtd|d|d wxYw#|:tjtj|tjdwwxYw) NrzThe repository for z contains custom code which must be executed to correctly load the model. You can inspect the repository content at https://hf.co/z. You can avoid this prompt in future by passing the argument `trust_remote_code=True`. Do you wish to run the custom code? [y/N] )yesy1T)non0FzS. Please pass the argument `trust_remote_code=True` to allow custom code to be run.zLoading z requires you to execute the configuration file in that repo on your local machine. Make sure you have read the code there to avoid malicious use, then set the option `trust_remote_code=True` to remove this error.) TIME_OUT_REMOTE_CODEsignalSIGALRMrlalarminputlower Exceptionri)trust_remote_code model_namehas_remote_codeprev_sig_handleranswers rresolve_trust_remote_coders} 3a7#  $#)==AU#V  12'/"-j\:ccmbnoEFF ||~)::,0)+??,1)(/ Q$/MM&..2BCLLO  t ,0zl#M M  -  )*6__i^jkhi $/MM&..2BCLLO0sBDDD##D&&>E$ctjj|}|jdr|dd}|j tjj d}t t|z }t5|r4tjj|dtjtjj|}tjj!||}|3tjj#|}|tj|<n|}|j$j'|ddd| t)St+|S#1swY"xYw)zY Import a module on the cache directory for modules and extract a class from it. rLNrY)location)r.r,normpathendswithreplaceseprr_HF_REMOTE_CODE_LOCKr+modulespopr_invalidate_cachesgetutilspec_from_file_locationmodule_from_specloader exec_modulefind_pipeline_classgetattr) class_namerT force_reloadr4rF cached_module module_specmodules rget_class_in_modulers 77  K (D }}UCRy << S )D-. You may pass a token in `token` if you are not logged in (`hf auth login`) and want to use private or [gated models](https://huggingface.co/docs/hub/models-gated#gated-models). Returns: `str`: The path to the module inside the cache. local/rvrYNrmainzDefaulting to latest_version: z`custom_revision`: z3 does not exist. Please make sure to choose one of rZdatasetrL)repo_id repo_typercrrrrgitz Revision 'a' not found in the community pipelines mirror. Check available revisions on https://huggingface.co/datasets/diffusers/community-pipelines-mirror/tree/main. If you don't find the revision you are looking for, please open an issue on https://github.com/huggingface/diffusers/issues.zCould not locate the z inside )rrrrrz--r)rrrrrrrr)rNr.r,rbisfilecountr&rr^loggerinforir COMMUNITY_PIPELINES_MIRROR_IDr EnvironmentErrorerrorrgrrr7rrshutilcopyfilerOr0r/r shaget_cached_module_file)rrFrrrrrrmodule_file_or_urlresolved_module_file submoduleavailable_versionslatest_versionemodules_neededfull_submodulesubmodule_path module_needed module_folder commit_hashs rrrsr%((E$F!&C[Q ww~~()1 & , ,S 1Q 635sxx (9(9#(>r(BCC  )7);?Q)Q~W]H KK8 !D E + +8*~H  H%hZ0II0F8;<=Q@   #25#$:Q'D&ESI#-!1$ I7%?K #2-#-!1$  Wdii8U8[8[\_8`.abI ##78N3RWW[[@9LN.)*+n?,oS1M OOBGGLL)F VXfivXv w x!!>Y^_cc (+5'"''++5 Cn-,446;$$S)*a/ + 1 1# 6q 9 ww~~n}&DEKK >? OO0.;2N O, M=&&s+,1 #)<) Calling this function will execute the code in the module file found locally or downloaded from the Hub. It should therefore only be called on trusted repos. Args: pretrained_model_name_or_path (`str` or `os.PathLike`): This can be either: - a string, the *model id* of a pretrained model configuration hosted inside a model repo on huggingface.co. Valid model ids can be located at the root-level, like `bert-base-uncased`, or namespaced under a user or organization name, like `dbmdz/bert-base-german-cased`. - a path to a *directory* containing a configuration file saved using the [`~PreTrainedTokenizer.save_pretrained`] method, e.g., `./my_model_directory/`. module_file (`str`): The name of the module file containing the class to look for. class_name (`str`): The name of the class to import in the module. cache_dir (`str` or `os.PathLike`, *optional*): Path to a directory in which a downloaded pretrained model configuration should be cached if the standard cache should not be used. force_download (`bool`, *optional*, defaults to `False`): Whether or not to force to (re-)download the configuration files and override the cached versions if they exist. proxies (`Dict[str, str]`, *optional*): A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}.` The proxies are used on each request. token (`str` or `bool`, *optional*): The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated when running `transformers-cli login` (stored in `~/.huggingface`). revision (`str`, *optional*, defaults to `"main"`): The specific model version to use. It can be a branch name, a tag name, or a commit id, since we use a git-based system for storing models and other artifacts on huggingface.co, so `revision` can be any identifier allowed by git. local_files_only (`bool`, *optional*, defaults to `False`): If `True`, will only try to load the tokenizer configuration from local files. You may pass a token in `token` if you are not logged in (`hf auth login`) and want to use private or [gated models](https://huggingface.co/docs/hub/models-gated#gated-models). Returns: `type`: The class, dynamically imported from the module. Examples: ```python # Download module `modeling.py` from huggingface.co and cache then extract the class `MyBertModel` from this # module. cls = get_class_from_dynamic_module("sgugger/my-bert-model", "modeling.py", "MyBertModel") ```r)rr) rrFrrrrrrrkwargs final_modules rget_class_from_dynamic_modulers6T*%%) L z< 88r)F)NFNNNF)NNFNNNF)9__doc__r_rrr.rArrvr+ threadingpathlibrtypesrtypingrrrurllibrhuggingface_hubr r huggingface_hub.utilsr r packagingr rtrrrr get_loggerrrrintgetenvruLockrr&r3rNPathLiker7rJrWrgrlrrrboolrrrrrsd: ((7MFF   H %!G9299%DbIJ%y~~'>  c2;;&6 7 '$ :*@)X'B648 (,(,""m5#(bkk)9#:m5m5c2;;./0m5 m5 d38n % m5 E$)$ % m5smm5m5m5`!%37 (,(,""S9#(bkk)9#:S9S9 S9c2;;./0 S9  S9 d38n % S9 E$)$ %S9smS9S9S9r