Pi.1ddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZGd deZdS)N) HTTPStatus)Path)warn)snapshot_download)GatedRepoErrorRepositoryNotFoundError)model_download)set_kaggle_credentials)KaggleApiHTTPError)parse_model_handle) Subcommand) REPO_ID_FNAMEceZdZdZdejffd ZddZdejddfdZ dejddfd Z dejddfd Z d e ddfd Z dejfd ZxZS)Downloadz7Holds all the logic for the `tune download` subcommand. subparsersc *t|dddddtjdt j|_||j |j dS)Ndownloadz tune downloadz!tune download [OPTIONS]z?Download a model from the Hugging Face Hub or Kaggle Model Hub.a examples: # Download a model from the Hugging Face Hub with a Hugging Face API token $ tune download meta-llama/Llama-2-7b-hf --hf-token Successfully downloaded model repo and wrote to the following locations: /tmp/Llama-2-7b-hf/config.json /tmp/Llama-2-7b-hf/README.md /tmp/Llama-2-7b-hf/consolidated.00.pth ... # Download an ungated model from the Hugging Face Hub $ tune download mistralai/Mistral-7B-Instruct-v0.2 --output-dir /tmp/model Successfully downloaded model repo and wrote to the following locations: /tmp/model/config.json /tmp/model/README.md /tmp/model/model-00001-of-00002.bin ... # Download a model from the Kaggle Model Hub $ tune download metaresearch/llama-3.2/pytorch/1b --source kaggle Successfully downloaded model repo and wrote to the following locations: /tmp/llama-3.2/pytorch/1b/tokenizer.model /tmp/llama-3.2/pytorch/1b/params.json /tmp/llama-3.2/pytorch/1b/consolidated.00.pth ... For a list of all models, visit the Hugging Face Hub https://huggingface.co/models or Kaggle Model Hub https://kaggle.com/models. )progusagehelp descriptionepilogformatter_class)func) super__init__ add_parsertextwrapdedentargparseRawTextHelpFormatter_parser_add_arguments set_defaults _download_cmd)selfr __class__s k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchtune/_cli/download.pyrzDownload.__init__ s !,,  5RY?>%9K-& &  N  !!t'9!:::::returnNc  |jdtd|jdtddd|jd tdt jd dd |jd tdd |jdtddddgd|jdtdd|jdtdddS)zAdd arguments to the parser.repo_idzOName of the repository on Hugging Face Hub or model handle on Kaggle Model Hub.)typerz --output-dirFNzFDirectory in which to save the model. Defaults to `/tmp/`.)r-requireddefaultrz --hf-tokenHF_TOKENzrn transformerszRequested model zc is neither a PyTorch nor a Transformers model, and therefore may not be compatible with torchtune.zFailed to validate z with error .)r frameworkownerrrQr"rP)r&rl parsed_handler]r_s r(rdz&Download._validate_kaggle_model_handles  $.v66M'944!'>99r)r)r rWr6rrRhttprpathlibrwarningsrhuggingface_hubrhuggingface_hub.utilsrr kagglehubr kagglehub.authr kagglehub.exceptionsr kagglehub.handler torchtune._cli.subcommandr 'torchtune.training.checkpointing._utilsrrr)r(rsA ------IIIIIIII$$$$$$111111333333//////000000AAAAAABBBBBzBBBBBr)