U ʹjv @sdZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZd d lmZe eZGd d d e ZdS)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)ListOptional)logging)snapshot_download)BaseHuggingfaceCLICommand)hf_hub_download)disable_progress_barsenable_progress_bars)show_deprecation_warningc@sJeZdZeedddZeddddZddd d Ze dd d Z dS) DownloadCommand)parsercCs|jddd}|jdtdd|jdtdd d |jd d d dgd dd|jdtdd|jddtdd|jddtdd|jdtdd|jdtdd|jddddgd d!|jd"d#d$d%|jd&d#d'd%|jd(td)d|jd*d#d+d%|jd,td-d.d/|jtd0dS)1NdownloadzDownload files from the Hub)helprepo_idzFiles to download (e.g. `config.json`, `data/metadata.jsonl`).)rnargsrz --repo-typemodelZdatasetspacez4Type 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. Check out https://huggingface.co/docs/huggingface_hub/guides/download#download-files-to-local-folder for more details.z--local-dir-use-symlinksautoTrueFalsezWDeprecated and ignored. Downloading to a local directory does not use symlinks anymore.)rrz--force-download store_truezFIf True, the files will be downloaded even if they are already cached.)actionrz--resume-downloadzDeprecated and ignored. Downloading a file to local dir always attempts to resume previously interrupted downloads (unless hf-transfer is enabled).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.z --max-workerszEMaximum number of workers to use for downloading files. Default is 8.)rrr)func) add_parser add_argumentstrint set_defaultsr)rZdownload_parserr'`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/huggingface_hub/commands/download.pyregister_subcommand8s z#DownloadCommand.register_subcommandN)argsreturncCs|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j |_ |j pXd|_ |j |_ |j |_ |j dk rtdtdS)NzbIgnoring --local-dir-use-symlinks. Downloading to a local directory does not use symlinks anymore.)tokenrr repo_typerevisionincludeexclude cache_dir local_dirforce_downloadresume_downloadquiet max_workersZlocal_dir_use_symlinkswarningswarn FutureWarning)selfr*r'r'r(__init__}s$  zDownloadCommand.__init__)r+c Cshtdd|jrHtttdt|W5QRXtnt t|t dS)Nzhuggingface-cli downloadz hf downloadignore) r r5r r7catch_warnings simplefilterprint _downloadr rZset_verbosity_infoZset_verbosity_warning)r:r'r'r(runs    zDownloadCommand.runcCst|jdkrR|jdk r0t|jdkr0td|jdk rRt|jdkrRtdt|jdkrt|j|j|j |jd|j |j |j |j |jdd St|jdkr|j}|j}n |j}d}t|j|j|j |||j |j |j |j |jd|jd S)Nrz?Ignoring `--include` since filenames have being explicitly set.z?Ignoring `--exclude` since filenames have being explicitly set.r zhuggingface-cli) rr-r.filenamer1r4r3r,r2 library_name) rr-r.allow_patternsignore_patternsr4r3r1r,r2rCr6)lenrr/r7r8r0r rr-r.r1r4r3r,r2rr6)r:rDrEr'r'r(r@sJ  zDownloadCommand._download) __name__ __module__ __qualname__ staticmethodrr)rr;rAr$r@r'r'r'r(r7s Dr)__doc__r7argparserrtypingrrZhuggingface_hubrZ"huggingface_hub._snapshot_downloadrZhuggingface_hub.commandsrZhuggingface_hub.file_downloadr Zhuggingface_hub.utilsr r Z _cli_utilsr Z get_loggerrGloggerrr'r'r'r(s