o Zh#@sdZddlZddlmZmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZe eZGd d d eZdS) aContains command to download files from the Hub with the CLI. Usage: huggingface-cli download --help # Download file huggingface-cli download gpt2 config.json # Download entire repo huggingface-cli download fffiloni/zeroscope --repo-type=space --revision=refs/pr/78 # Download repo with filters huggingface-cli download gpt2 --include="*.safetensors" # Download with token huggingface-cli download Wauplin/private-model --token=hf_*** # Download quietly (no progress bar, no warnings, only the returned path) huggingface-cli download gpt2 config.json --quiet # Download to local dir huggingface-cli download gpt2 --local-dir=./models/gpt2 N) Namespace_SubParsersAction)ListLiteralOptionalUnion)logging)snapshot_download)BaseHuggingfaceCLICommand)HF_HUB_ENABLE_HF_TRANSFER)hf_hub_download)disable_progress_barsenable_progress_barsc@sHeZdZedefddZdeddfddZd d d Zde fd d Z dS)DownloadCommandparsercCs|jddd}|jdtdd|jdtdd d |jd gd d dd|jdtdd|jddtdd|jddtdd|jdtdd|jdtdd|jdgdddd|jdd d!d"|jd#d d$d"|jd%td&d|jd'd d(d"|jtd)dS)*NdownloadzDownload files from the Hub)helprepo_idzFiles to download (e.g. `config.json`, `data/metadata.jsonl`).)rnargsrz --repo-type)modelZdatasetspacerz4Type of repo to download from (defaults to 'model').)choicesdefaultrz --revisionzPAn optional Git revision id which can be a branch name, a tag, or a commit hash.z --includez)Glob patterns to match files to download.)rrrz --excludez0Glob patterns to exclude from files to download.z --cache-dirz9Path to the directory where to save the downloaded files.z --local-dirzIf set, the downloaded file will be placed under this directory either as a symlink (default) or a regular file. Check out https://huggingface.co/docs/huggingface_hub/guides/download#download-files-to-local-folder for more details.z--local-dir-use-symlinks)autoTrueFalserawTo be used with `local_dir`. If set to 'auto', the cache directory will be used and the file will be either duplicated or symlinked to the local directory depending on its size. It set to `True`, a symlink will be created, no matter the file size. If set to `False`, the file will either be duplicated from cache (if already exists) or downloaded from the Hub and not cached.z--force-download store_truezFIf True, the files will be downloaded even if they are already cached.)actionrz--resume-downloadz2If True, resume a previously interrupted download.z--tokenzIA User Access Token generated from https://huggingface.co/settings/tokensz--quietzWIf True, progress bars are disabled and only the path to the download files is printed.)func) add_parser add_argumentstr set_defaultsr)rZdownload_parserr&X/var/www/html/lang_env/lib/python3.10/site-packages/huggingface_hub/commands/download.pyregister_subcommand7sn  z#DownloadCommand.register_subcommandargsreturnNcCs|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j |_ |j |_ |j |_ ||j }|dkr@d|_ dS|dkrId|_ dS|dkrRd|_ dStd|j d)NtrueTfalseFr'za' is not a valid value for `local_dir_use_symlinks`. It must be either 'auto', 'True' or 'False'.)tokenrr repo_typerevisionincludeexclude cache_dir local_dirforce_downloadresume_downloadquietlocal_dir_use_symlinkslower ValueError)selfr)Zuse_symlinks_lowercaser&r&r'__init__{s.     zDownloadCommand.__init__cCst|jr*tttdt|Wdn1s wYtdSt t|t dS)Nignore) r7r warningscatch_warnings simplefilterprint _downloadrrZset_verbosity_infoZset_verbosity_warning)r;r&r&r'runs     zDownloadCommand.runcCst|jdkr)|jdurt|jdkrtd|jdur)t|jdkr)tdts0tdt|jdkrRt |j |j |j |jd|j |j|j|j|j|jdd St|jdkr`|j}|j}n|j}d}t|j |j |j |||j|j|j |j|j|jdd S) Nrz?Ignoring `--include` since filenames have being explicitly set.z?Ignoring `--exclude` since filenames have being explicitly set.zConsider using `hf_transfer` for faster downloads. This solution comes with some limitations. See https://huggingface.co/docs/huggingface_hub/hf_transfer for more details.zhuggingface-cli) rr/r0filenamer3r6r5r.r4r8 library_name) rr/r0allow_patternsignore_patternsr6r5r3r.r4r8rF)lenrr1r>warnr2r loggerinfor rr/r0r3r6r5r.r4r8r )r;rGrHr&r&r'rBsT  zDownloadCommand._download)r*N) __name__ __module__ __qualname__ staticmethodrr(rr<rCr$rBr&r&r&r'r6s C  r)__doc__r>argparserrtypingrrrrZhuggingface_hubrZ"huggingface_hub._snapshot_downloadr Zhuggingface_hub.commandsr Zhuggingface_hub.constantsr Zhuggingface_hub.file_downloadr Zhuggingface_hub.utilsr rZ get_loggerrMrKrr&r&r&r's