a h@sdZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZe eZGd d d e ZdS) a_Contains command to download files from the Hub with the CLI. Usage: hf download --help # Download file hf download gpt2 config.json # Download entire repo hf download fffiloni/zeroscope --repo-type=space --revision=refs/pr/78 # Download repo with filters hf download gpt2 --include="*.safetensors" # Download with token hf download Wauplin/private-model --token=hf_*** # Download quietly (no progress bar, no warnings, only the returned path) hf download gpt2 config.json --quiet # Download to local dir hf download gpt2 --local-dir=./models/gpt2 N) Namespace_SubParsersAction)ListOptional)logging)snapshot_download)BaseHuggingfaceCLICommand)hf_hub_download)disable_progress_barsenable_progress_barsc@sJeZdZeedddZeddddZddd d Ze dd d Z dS) DownloadCommand)parsercCs|jddd}|jdtdd|jdtdd d |jd gd d dd|jdtdd|jddtdd|jddtdd|jdtdd|jdtdd|jdddd|jdtd d|jd!dd"d|jd#td$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. Check out https://huggingface.co/docs/huggingface_hub/guides/download#download-files-to-local-folder for more details.z--force-download store_truezFIf True, the files will be downloaded even if they are already cached.)actionrz--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 )r Zdownload_parserr"X/var/www/html/assistant/venv/lib/python3.9/site-packages/huggingface_hub/cli/download.pyregister_subcommand6sh z#DownloadCommand.register_subcommandN)argsreturncCsd|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j |_ |j |_ |j |_ dS)N) tokenrr repo_typerevisionincludeexclude cache_dir local_dirforce_downloadquiet max_workers)selfr%r"r"r#__init__qszDownloadCommand.__init__)r&cCsr|jrRtt&tdt|Wdn1s@0Ytnt t|t dS)Nignore) r/r warningscatch_warnings simplefilterprint _downloadr rZset_verbosity_infoZset_verbosity_warning)r1r"r"r#runs  * zDownloadCommand.runc Cst|jdkrR|jdur0t|jdkr0td|jdurRt|jdkrRtdt|jdkrt|j|j|j |jd|j |j |j |j dd St|jdkr|j}|j}n |j}d}t|j|j|j |||j |j |j |j d|jd S)Nrz?Ignoring `--include` since filenames have being explicitly set.z?Ignoring `--exclude` since filenames have being explicitly set.Zhf) rr(r)filenamer,r.r'r- library_name) rr(r)allow_patternsignore_patternsr.r,r'r-r<r0)lenrr*r4warnr+r rr(r)r,r.r'r-rr0)r1r=r>r"r"r#r8sF   zDownloadCommand._download) __name__ __module__ __qualname__ staticmethodrr$rr2r9rr8r"r"r"r#r 5s : r )__doc__r4argparserrtypingrrZhuggingface_hubrZ"huggingface_hub._snapshot_downloadrZhuggingface_hub.commandsrZhuggingface_hub.file_downloadr Zhuggingface_hub.utilsr r Z get_loggerrAloggerr r"r"r"r#s