a  d < @sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZddlZddlmZmZmZzddlmZWneyddlmZYn0zddlZd ZWneyd ZYn0ejd krdd l mZn dd lmZdd lmZddlm Z zBddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ee$dedZ$d Z*WneyndZ$d Z*Yn0e+e,Z-gdZ.dZ/dZ0dZ1dZ2d@ddZ3dAddZ4dBddZ5dCdd Z6e7d!d"d#Z8ee7ej9fd$d%d&Z:e7e7d'd(d)Z;e7d*d+d,Zd/d0d1Z?dEe7e e>ee@ed2fd3d4d5ZAdFe7e7e e7e e7e@e@e e>e e>ee@ed2fd7 d8d9ZBe>e7d:d;d<ZCe7e e7d=d>d?ZDdS)GN)partial)Path)TemporaryDirectory)IterableOptionalUnion) HASH_REGEXdownload_url_to_fileurlparse)get_dir)_get_torch_homeTF))Literal) __version__)filter_pretrained_cfg) create_repoget_hf_file_metadatahf_hub_download hf_hub_urlrepo_type_and_id_from_hf_id upload_folder)EntryNotFoundErrortimm)Z library_nameZlibrary_version) get_cache_dirdownload_cached_file has_hf_hubhf_splitload_model_config_from_hfload_state_dict_from_hf save_for_hfpush_to_hf_hubzpytorch_model.binzmodel.safetensorszopen_clip_pytorch_model.binzopen_clip_model.safetensorscCsPtdrtdt}|s"dn|f}tjj|dg|R}tj|dd|S)zf Returns the location of the directory where models are cached (and creates it if necessary). ZTORCH_MODEL_ZOOz@TORCH_MODEL_ZOO is deprecated, please use env TORCH_HOME insteadZ checkpointsT)exist_ok)osgetenv_loggerwarningr pathjoinmakedirs)Z child_dirZhub_dirZ model_dirr#r#Y/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/models/_hub.pyr9s  rcCst|ttfr|\}}nt|}tj|j}tjt|}tj |st d ||d}|rt |}|r||dnd}t||||d|S)NzDownloading: "{}" to {} )progress) isinstancelisttupler r%r)basenamer*rexistsr'infoformatrsearchgroupr )url check_hashr.filenameparts cached_file hash_prefixrr#r#r,rHs   rc Cst|ttfr|\}}nt|}tj|j}tjt|}tj |r|rt |}|rf| dnd}|rt |dF}t|}|dt||krWddSWdn1s0YdSdS)Nr-rbFT)r/r0r1r r%r)r2r*rr3rr6r7openhashlibsha256read hexdigestlen) r8r9r:r;r<r>r=fZhdr#r#r,check_cached_fileYs     .rGcCsts|rtdtS)Nz^Hugging Face hub model specified but package not installed. Run `pip install huggingface_hub`.) _has_hf_hub RuntimeError)Z necessaryr#r#r,rms r)hf_idcCsT|d}dt|kr"dks,nJd|d}t|dkrH|dnd}||fS)N@rzChf_hub id should only contain one @ character to identify revision.r-)splitrE)rJZ rev_split hf_model_id hf_revisionr#r#r,rus  "r) json_filecCs@t|ddd}|}Wdn1s,0Yt|S)Nr>zutf-8)encoding)r@rCjsonloads)rQreadertextr#r#r,load_cfg_from_json~s&rW)model_idr:cCst|\}}t|||dS)N)revision)rr)rXr:rOrPr#r#r,download_from_hfs rZ)rXcCstds Jt|d}t|}d|vrj|}i}|d|d<|dd|d<d|vrb|d|d<||d<|d}||d<d |d <d |vr|d |d <d|vr|d|d<d |vr|d |d <|d}||fS) NT config.jsonpretrained_cfg architecture num_featureslabels label_namesZ hf_hub_idzhf-hubsource num_classeslabel_descriptions)rrZrWpop)rXr< hf_configr\ model_namer#r#r,rs,   rc Cstds Jt|\}}trt|D]Z}zBt|||d}td|d|d|dtjj |ddWSt y|Yq$0q$t|||d }t d|d |d tj |dd S) NTrepo_idr:rY[z"] Safe alternative available for 'z' (as 'z&'). Loading weights using safetensors.cpu)Zdevice)r:rYz"] Safe alternative not found for 'z)'. Loading weights using default pytorch.)Z map_location) rr_has_safetensors_get_safe_alternativesrr'r4 safetensorstorchZ load_filerdebugload)rXr:rOrPZ safe_filenameZcached_safe_filer<r#r#r,rs"    r) config_path model_configc Cs@|pi}i}t|jddd}|d|d<|d|j|d<|d|j|d<|dt|dd}t|trx|rx||d<d|vrt d| d |d|d d}|rt|t t tfsJ||d <|d d}|rt|t sJ||d <||d <|||d  }tj||d dWdn1s20YdS)NT)Z remove_sourceZ remove_nullr]rbr^Z global_poolr_z'labels' as a config field for is deprecated. Please use 'label_names' and 'label_descriptions'. Renaming provided 'labels' field to 'label_names'.r`rcr\wrL)indent)rr\rdgetrbr^getattrr/strr'r( setdefaultdictr0r1updater@rSdump) modelrqrrrer\Zglobal_pool_typer`rcrFr#r#r,save_config_for_hfs4    r}both)save_directoryrrsafe_serializationcCstds Jt|}|jddd|}|dus:|dkrXtsFJdtj||t|dush|dkrxt ||t |d}t |||ddS)NT)r$parentsr~z-`pip install safetensors` to use .safetensorsFr[)rr) rrmkdirZ state_dictrkrmrnZ save_fileHF_SAFE_WEIGHTS_NAMEsaveHF_WEIGHTS_NAMEr})r|rrrrZtensorsrqr#r#r,r s  r Add model) rhcommit_messagetokenrYprivate create_prrr model_cardrc Cst|||dd} t| \} } } | d| }ztt|d|dd}Wnty\d}Yn0tl}t|||| d|s|pi}|dd}t|d}t ||}| |t |||||d Wd S1s0Yd S) a5 Arguments: (...) safe_serialization (`bool` or `"both"`, *optional*, defaults to `False`): Whether to save the model using `safetensors` or the traditional PyTorch way (that uses `pickle`). Can be set to `"both"` in order to push both safe and unsafe weights. T)rrr$/z README.mdrgF)rrrrM)rhZ folder_pathrYrrN) rrrrrrr rNrgenerate_readme write_textr)r|rhrrrYrrrrrrrepo_url_Z repo_owner repo_nameZ has_readmetmpdirrfZ readme_path readme_textr#r#r,r!s.     r!)rrfc Csd}|d7}|d7}|d|ddd7}d|vrd |dvr|d 7}t|dd ttfr|dd D]}|d |d7}qnn|d |dd d7}d |dvrt|dd ttfr|dd D]}|d |d7}qn|d |dd d7}|d7}|d |d7}d|vrD|d|dd7}d|vr |d7}|dD]\}}t|ttfr|d|d7}|D]}|d|d7}qn^t|tr|d|d7}|D] \}}|d|d|d7}qn|d|d|d7}qbd|vr0|d7}||d7}|d7}d|vrV|d7}||d7}|d7}d|vr|d7}t|dttfs|dg}n|d}|D]} |d| d7}q|S)Nz--- z$tags: - image-classification - timm zlibrary_name: timm z license: licensez apache-2.0 detailsZDatasetz datasets: z- zPretrain Datasetz# Model card for descriptionz ## Model Details z- **z:** z - z: z:** usagez ## Model Usage Z comparisonz ## Model Comparison Zcitationz ## Citation z ```bibtex z ``` )rur/r1r0loweritemsry) rrfrdkvviZkiZ citationscr#r#r,r:s^         r)r:returnccs:|tkrtV|ttfvr6|dr6|dddVdS)aIReturns potential safetensors alternatives for a given filename. Use case: When downloading a model from the Huggingface Hub, we first look if a .safetensors file exists and if yes, we use it. Main use case is filename "pytorch_model.bin" => check for "model.safetensors" or "pytorch_model.safetensors". z.binNz .safetensors)rrHF_OPEN_CLIP_WEIGHTS_NAMEendswith)r:r#r#r,rlrsrl)r")TF)T)F)N)NF)rNNFFNNF)ErArSloggingr%sys functoolsrpathlibrtempfilertypingrrrrnZ torch.hubrr r r ImportErrorr Zsafetensors.torchrmrk version_inforZtyping_extensionsrrZtimm.models._pretrainedrZhuggingface_hubrrrrrrZhuggingface_hub.utilsrrH getLogger__name__r'__all__rrrZHF_OPEN_CLIP_SAFE_WEIGHTS_NAMErrrGrrwrPathLikerWrZrrryr}boolr r!rrlr#r#r#r,s                   " -  98