rh؇VdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZmZmZddlmZddlmZdd lmZmZmZmZmZm Z dd l!m"Z"m#Z#e jHe%Z&e jNZ(d Z)d ee*ejVfd dfdZ,dee*ejVfd e-e*fdZ.dee*ejVfd e-e*fdZ/dee*ejVfd e-e*fdZ0dee*ejVfd e-e*fdZ1ddde*dee*ejVfde2d e3fdZ4 d1dee*ejVfde*deee*ejVfde2dee2dee5e*e*fd eee2e*fd!ee*d"e2d#ee*d$ee*d e*fd%Z6 d1d&e*dee*ejVfdeee*ejVfde2dee2dee5e*e*fd eee2e*fd!ee*d"e2d#ee*d'ee*d e3fd(Z7d2d)ed*ee*ejVfd+ee5d e-e*fd,Z8d-Z9d.Z: d3d/Z;d4d0ZQCs)>>()Q. ##N3 GY>sB'! B,+B,5 B1filenamect|d5}|j}dddtddlfdt j }|t S#1swYFxYw)a Extracts all the libraries (not relative imports this time) that are imported in a file. Args: filename (`str` or `os.PathLike`): The module file to inspect. Returns: `list[str]`: The list of all packages required to use the input module. r-r.Nrcvt|tjryt|tjr|j}tj |D]x}t|tj st|jdd}|jdr|jds!t jj|sxynt|tjrD|jD]4}|j j#dd}|s$j%|6nft|tj&rL|j(dk(r=|j*r1|j*j#dd}|rj%|tj,|D] }| y)Nid available is_flash_attn.r) isinstanceastTryIftestwalkCallgetattrfuncendswith startswithhasattrutils import_utilsImportnamesr$splitadd ImportFromlevelmoduleiter_child_nodes) noderUcondition_nodecheck_functionalias top_modulechildimported_modulesrecursive_look_for_imports transformerss r!rnz/get_imports..recursive_look_for_importssW dCGG $  cff %99D"%((4. nchh7%,^-@-@$%KN&// <*55oF"<#5#5#B#BNS cjj ) 5"ZZ--c215 $((4 5cnn -zzQ4;;![[..s3A6 $((4))$/ .E &u - .r#)r1r2r7transformers.utilsrRparsesorted)rIr8r9treermrnros @@@r! get_importsrtsg h )Q&&(u.B 99W Dt$ " ##Us A''A0ct|}g}|D]} tj|t|dkDr0tddj|dd j|d t|S#t$rE}tj d|d|dt |vr|j|nYd}~d}~wwxYw) a. Check if the current Python environment contains all the libraries that are imported in a file. Will raise if a library is missing. Args: filename (`str` or `os.PathLike`): The module file to check. Returns: `list[str]`: The list of relative imports in the file. z&Encountered exception while importing z: zNo module namedNrz\This modeling file requires the following packages that were not found in your environment: z, z. Run `pip install  `) rtr import_module ImportErrorloggerwarningr?rr@joinr;)rIimportsmissing_packagesimp exceptions r! check_importsrs(#G    # #C (  q  jyy)*++>sxxHX?Y>ZZ[ ]   ))!  NNCC59+V W!C N2 '',-  sA44 C=;B==CF force_reload class_namerErc tjj|}|jdr|dd}|j tjj d}t t|z }t5|r4tjj|dtjtjj|}tjj!||}|gt#t%t t'|z}t)j*dj-d|Dj/}|3tjj1|} | tj|<n|} t3| dd |k7r"|j4j7| || _t3| |cdddS#1swYyxYw) a Import a module on the cache directory for modules and extract a class from it. Args: class_name (`str`): The name of the class to import. module_path (`str` or `os.PathLike`): The path to the module to import. force_reload (`bool`, *optional*, defaults to `False`): Whether to reload the dynamic module from file if it already exists in `sys.modules`. Otherwise, the module is only reloaded if the file has changed. Returns: `typing.Type`: The class looked for. r=NrP)locationr#c3TK|] }t||jz"ywN)bytes read_bytes).0r8s r! z&get_class_in_module.. s 2cQR58alln3L2cs&(__transformers_module_hash__rM)rrnormpathrZreplaceseprr _HF_REMOTE_CODE_LOCKrmodulespoprrgetutilspec_from_file_locationrrmaprHhashlibsha256r| hexdigestmodule_from_specrXloader exec_moduler) rrErr$r+ cached_module module_spec module_files module_hashres r!get_class_in_modulersj& 77  K (D }}UCRy << S )D-.>#((2cVb2c*cdnnp   ^^44[AF &CKK "F 692 >+ M    * *6 22=F /vz*-+++s 8E G  Gpretrained_model_name_or_path cache_dirforce_downloadresume_downloadproxiestokenrevisionlocal_files_only repo_type _commit_hashc | jdd} | )tjdt| t d| }t r|st jdd}t|}tjj|} | r tjj|}n:|jdtjj}t|||| | }g} t!|||||||||| | }| s|k7r|j#|t'|}t(tjjz|z}t+|t-t.|z }|tjj|k(r(||z j1r"t3j4|t||z sM||z j6j9dd t;j<|||z t?j@|D]}t-|j6|dz }tjjC||}||z j1r#t3j4|t||z rtt;j<|||z t?j@ntE|| }||z }|tjjz|z}tjjC||}t+t-|j6||z j1s-t;j<|||z t?j@|D]E}||dz j1rtG||d|||||||| |j#|dGtI|dkDrP|NdjC|Dcgc]}d| c}}| dn| d}d||}t jKd|d|dtjjC||S#t$$rt jd |d |d wxYwcc}w)a Prepares Downloads a module from a local folder or a distant repo and returns its path inside the cached Transformers module. 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. - 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. 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. resume_download: Deprecated and ignored. All downloads are now resumed by default when possible. Will be removed in v5 of Transformers. 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 `hf auth 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. repo_type (`str`, *optional*): Specify the repo type (useful when downloading from a space for instance). Passing `token=True` is required when you want to use a private model. Returns: `str`: The path to the module inside the cache. use_auth_tokenNrThe `use_auth_token` argument is deprecated and will be removed in v5 of Transformers. Please use `token` instead.V`token` and `use_auth_token` are both specified. Please set only the argument `token`.z+Offline mode: forcing local_files_only=TrueT/)rrr) rrrrrrrrrzCould not locate the z inside rP)parentsrr=)rrrrrrrrr z- rMzs/zhttps://huggingface.co/z9A new version of the following files was downloaded from z: z . Make sure to double-check they do not contain any added malicious code. To avoid downloading new versions of the code file, you can pin a revision.)&rwarningswarn FutureWarning ValueErrorrrzinfor?rrisdirbasenamerrrr rOSErrorrr r)rr rfilecmpcmpr(mkdirshutilcopyrrr|rget_cached_module_filer@r{)rr+rrrrrrrrrdeprecated_kwargsris_local submoduler new_filesresolved_module_filemodules_neededfull_submodulesubmodule_path module_neededmodule_needed_file commit_hashfull_submodule_module_file_pathr8 repo_type_strurls r!rrs;|'**+;TBN!  A    uv v!1 AB%((E$F!ww}}:;HGG$$%BC 199#rww{{K . );)Vbnw I* ) )+-% M-AA   [ )##78N6 CiON.)*+n M (+5'"''++5 C*,'',,~{*S'd#BCJJK,446 KK,n{.J K  ' ' )+ 8M" c%::BBD&1$oS)'#1$3#%%5!,   M?#!67 8  9~h.II;AA3x;< '/ {"5E ' 7T6UVGuCPY{ A A 77<< 44K  +K=A^@__`ab x 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: class_reference (`str`): The full name of the class to load, including its module and optionally its repo. 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. - a path to a *directory* containing a configuration file saved using the [`~PreTrainedTokenizer.save_pretrained`] method, e.g., `./my_model_directory/`. This is used when `class_reference` does not specify another repo. 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. resume_download: Deprecated and ignored. All downloads are now resumed by default when possible. Will be removed in v5 of Transformers. 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 `hf auth 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. repo_type (`str`, *optional*): Specify the repo type (useful when downloading from a space for instance). code_revision (`str`, *optional*, defaults to `"main"`): The specific revision to use for the code on the Hub, if the code leaves in a different repository than the rest of the model. 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. Passing `token=True` is required when you want to use a private model. Returns: `typing.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("modeling.MyBertModel", "sgugger/my-bert-model") # Download module `modeling.py` from a given repo and cache then extract the class `MyBertModel` from this # module. cls = get_class_from_dynamic_module("sgugger/my-bert-model--modeling.MyBertModel", "sgugger/another-bert-model") ```rNrrz--rPr=)rrrrrrrrr)rrrrrrarr)rrrrrrrrrrrkwargsrrepo_idr+r final_modules r!get_class_from_dynamic_modulerstZZ 0$7N!  A    uv v #2#8#8#> /-33C8K!>'!I )e%') L z._set_auto_map_in_config_sqmm.. !'',R0 "m1S]]%;%;$<= ) ##' #' }}%%..v6*5a 8N8N7O'P$3 6=I%,S2H%IN+9+D+D(+?+E+Ec+J2+N(.B-C1^E\E\D]+^(+6a 8N8N7O'P$-/CDI gt ${{:r2H(1HS__ %"*GJ  Wj$ / ;09G  S__ - #;G r#)rrzr{rQr6tuplerr__file__rr$rrrrH) rrrrcfgresult object_file dest_file needed_files ` r!custom_object_saverIs ~~#.se4x@  <@&4-( )C #C ( )  ' F++cnn-66KV [ 1 6 67I KK Y' MM)1=! LD$5$:$:;  K+ i ! Mr#ctd)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.)r)signumframes r!_raise_timeout_errorrs  c r#c|^| d|d|d|d}nOtjj|r'd|dtjj|d}n d|d|d}||rd}n|rtd kDrd} t j t j t}t jt|=t|d |d }|jd vrd }n|jdvrd}|=t jd  |Ht j t j |t jd n|r tdd|r|s|st|d |d|S#t$rt|d |dwxYw#|:t j t j |t jd wwxYw)a Resolves the `trust_remote_code` argument. If there is remote code to be loaded, the user must opt-in to loading it. Args: trust_remote_code (`bool` or `None`): User-defined `trust_remote_code` value. model_name (`str`): The name of the model repository in huggingface.co. has_local_code (`bool`): Whether the model has local code. has_remote_code (`bool`): Whether the model has remote code. error_message (`str`, *optional*): Custom error message to display if there is remote code to load and the user didn't opt-in. If unset, the error message will be regarding loading a model with custom code. Returns: The resolved `trust_remote_code` value. NzThe repository z% references custom code contained in zm which must be executed to correctly load the model. You can inspect the repository content at https://hf.co/z . zt contains custom code which must be executed to correctly load the model. You can inspect the repository content at z contains custom code which must be executed to correctly load the model. You can inspect the repository content at https://hf.co/Frz9 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)non0rMzS. Please pass the argument `trust_remote_code=True` to allow custom code to be run.) rrrabspathTIME_OUT_REMOTE_CODEsignalSIGALRMralarminputlower Exceptionr)trust_remote_code model_namehas_local_codehas_remote_code error_message upstream_repoprev_sig_handleranswers r!resolve_trust_remote_coders.  $!*-RS`Rab,,9?$@  WW]]: &!*. ggooj9:$@ "*.,,6G c g} td ||d|}t|d5}|j}ddd|D]}|j}|r|j dr' t |\}}} tjj|} |E| Ctj|tj| tj| } nd} | r|j|d| d  |r!t!d |d d j#|zy#1swYxYw#t $r |}d\}} YwxYw#tjj$r|j|dY7wxYw#t$rYwxYw)a Tries to locate `requirements_file` in a local folder or repo, and confirms that the environment has all the python dependencies installed. Args: path_or_repo_id (`str` or `os.PathLike`): This can be either: - a string, the *model id* of a model repo on huggingface.co. - a path to a *directory* potentially containing the file. kwargs (`dict[str, Any]`, *optional*): Additional arguments to pass to `cached_file`. )path_or_repo_idrIrN#NNz (installed: None)Tz (installed: )z4Missing requirements in your local environment for `z`: r)r r1 readlinesstripr[rrrmetadatar PackageNotFoundErrorrr from_stringrqrryr|) rrequirements_filerfailed requirementsr8 requirement package_name delimiterversion_numberlocal_package_version is_satisfieds r!check_python_requirementsr+sF" "i?M^ibhi , $ );;=L )( UK%++-K+"8"8"= 7:OP[:\7 i  (1(:(:(B(B<(P% $)CG0<B?BSSW X[_[d[dek[l l  E ) ) 7* ,6) > 7 %%::   -?@A     spE4D1E4!D&1D<A E4E4D#E4&D96E48D99E4<1E1-E40E11E44 F?F) NFNNNNFNNrr)zrequirements.txt)=__doc__rRrrrimportlib.metadataimportlib.utilrr3rrr threadingrpathlibrtypesrtypingrrrhuggingface_hubr packagingr r]r r r rrrutils.import_utilsrr get_loggerrrzLockrr"r?PathLiker)r6r;rHrtrbooltyperrrrrrrrr+rr#r!r;s:  ''2I   H %%y~~' & &c2;;&6 7&D&,'eC,<&='$s)'* 5bkk1A+B tCy B4$%R[[ 014$d3i4$n *E#r{{"23 *S *N .+.+sBKK'(.+ .+  .+h48 &*(,(,""#"&o5#(bkk)9#:o5o5c2;;./0o5 o5 d^ o5 d38n % o5 E$)$ %o5smo5o5}o53-o5 o5j48 &*(,(,""##'zVzV#(bkk)9#:zVc2;;./0zV zV d^ zV d38n % zV E$)$ %zVsmzVzV}zVC=zV zVzICIsBKK/?)@I(SW.IdhildmIXgkRj5 r#