U ʹj=@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z ddlZddlmZmZmZzddlmZWn ek rddlmZYnXzddlZd ZWnek rd ZYnXzdd l mZWn"ek rdd lmZYnXdd lmZdd lmZzBddlm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'ee"dedZ"d Z(Wnek r|dZ"d Z(YnXe)e*Z+ddddddddgZ,dZ-dZ.dZ/dZ0d>ddZ1d?d dZ2d@d!d"Z3dAd#dZ4e5d$d%dZ6e e5ej7fd&d'd(Z8e5e5d)d*d+Z9e5d,d-dZ:e-fe5e5d)d.dZ;dBe5e e<e e<d/d0d1Z=dCe5e e<e e<e e>ed2fd3d4dZ?dDej@jAe5e5e e5e e5e>e>e e<e e<e e<e e>ed2fd6 d7dZBerAjsonloads)rPreadertextr!r!r*load_cfg_from_json}srV)model_idr8cCst|\}}t|||dS)N)revision)rr)rWr8rNrOr!r!r*download_from_hfs rY)rWcCstds tt|d}t|}d|krj|}i}|d|d<|dd|d<d|krb|d|d<||d<|d}||d<d |d <d |kr|d |d <d|kr|d|d<d |kr|d |d <|d i}|d}|||fS)NT config.jsonpretrained_cfg architecture num_featureslabels label_namesZ hf_hub_idzhf-hubsource num_classeslabel_descriptions model_args)rrMrYrVpopget)rWr: hf_configr[rc model_namer!r!r*rs.    c Cstds tt|\}}trt|D]\}zBt|||d}td|d|d|dtj j |ddWSt k r~Yq$Xq$t|||d }t d|d |d t j |dd S) NTrepo_idr8rX[z"] Safe alternative available for 'z' (as 'z&'). Loading weights using safetensors.cpu)Zdevice)r8rXz"] Safe alternative not found for 'z)'. Loading weights using default pytorch.)Z map_location)rrMr_has_safetensors_get_safe_alternativesrr%r2 safetensorstorchZ load_filerdebugload)rWr8rNrOZ safe_filenameZcached_safe_filer:r!r!r*rs   ) config_path model_configrcc Cs6|pi}i}t|jddd}|d|d<|d|j|d<|d|j|d<|dt|dd}t|trx|rx||d<d|krt d| d |d|d d}|rt|t t t fst||d <|d d}|rt|t st||d <|r||d <||d <|||d } tj|| ddW5QRXdS)NT)Z remove_sourceZ remove_nullr\rar]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_rbrcr[wrJ)indent)rr[rdrar]getattrr-strr%r& setdefaultdictr.r/rMupdater>rRdump) modelrrrsrcrfr[Zglobal_pool_typer_rbrDr!r!r*save_config_for_hfs8    r}both)save_directoryrsrcsafe_serializationcCstds tt|}|jddd|}|dks:|dkrXtsFtdtj||t |dksh|dkrxt ||t |d}t ||||ddS)NT)r"parentsr~z-`pip install safetensors` to use .safetensorsFrZ)rsrc) rrMrmkdirZ state_dictrlrnroZ save_fileHF_SAFE_WEIGHTS_NAMEsaveHF_WEIGHTS_NAMEr})r|rrsrcrZtensorsrrr!r!r*rs    Add model) r|ricommit_messagetokenrXprivate create_prrs model_cardrcrc  Cst|||dd} t| \} } }| d|}ztt|d|dd}Wntk r^d}YnXtl}t|||| | d|s|pi}|dd}t|d}t ||}| |t |||||d W5QRSQRXd 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.mdrhF)rsrcrrK)riZ folder_pathrXrrN) rrrrrrrrLrgenerate_readme write_textr)r|rirrrXrrrsrrcrrepo_url_Z repo_owner repo_nameZ has_readmeZtmpdirrgZ readme_path readme_textr!r!r*rs:     )rrgc Csd}|d7}|d7}|d|ddd7}d|krd |dkr|d 7}t|dd ttfr|dd D]}|d |d7}qnn|d |dd d7}d |dkrt|dd ttfr|dd D]}|d |d7}qn|d |dd d7}|d7}|d |d7}d|krD|d|dd7}d|kr |d7}|dD]\}}t|ttfr|d|d7}|D]}|d|d7}qn^t|tr|d|d7}|D] \}}|d|d|d7}qn|d|d|d7}qbd|kr0|d7}||d7}|d7}d|krV|d7}||d7}|d7}d|kr|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 ``` )rer-r/r.loweritemsry) rrgrdkvviZkiZ citationscr!r!r*rNs^         r)r8returnccsH|tkrtV|tkrtV|ttfkrD|drD|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_NAMEHF_OPEN_CLIP_SAFE_WEIGHTS_NAMEendswith)r8r!r!r*rms rm)r )TF)T)F)NN)NNF) rNNFFNNNF)Er?rRloggingr# functoolsrpathlibrtempfilertypingrrrroZ torch.hubrr r r ImportErrorr Zsafetensors.torchrnrlr typing_extensionsrrZtimm.models._pretrainedrZhuggingface_hubrrrrrrZhuggingface_hub.utilsrrF getLogger__name__r%__all__rrrrrrrErrwrPathLikerVrYrrryr}boolrnnModulerrrmr!r!r!r*s                # 4  @8