phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update download command (#9)

Browse source
This commit is contained in:
Lucain 2024-07-25 01:50:40 +02:00 • committed by GitHub
parent 17bd1d876c
commit 378a2077dd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_toolchain/cli/download.py
View file

@ -48,13 +48,13 @@ class Download(Subcommand):
self.parser.add_argument( self.parser.add_argument(
"repo_id", "repo_id",
type=str, type=str,
help="Name of the repository on Hugging Face Hub eg. llhf/Meta-Llama-3.1-70B-Instruct", help="Name of the repository on Hugging Face Hub eg. meta-llama/Meta-Llama-3.1-70B-Instruct",
) )
self.parser.add_argument( self.parser.add_argument(
"--hf-token", "--hf-token",
type=str, type=str,
required=False, required=False,
default=os.getenv("HF_TOKEN", None), default=None,
help="Hugging Face API token. Needed for gated models like Llama2. Will also try to read environment variable `HF_TOKEN` as default.", help="Hugging Face API token. Needed for gated models like Llama2. Will also try to read environment variable `HF_TOKEN` as default.",
) )
self.parser.add_argument( self.parser.add_argument(
@ -74,11 +74,9 @@ class Download(Subcommand):
true_output_dir = snapshot_download( true_output_dir = snapshot_download(
args.repo_id, args.repo_id,
local_dir=output_dir, local_dir=output_dir,
# "auto" will download to cache_dir and symlink files to local_dir
# avoiding unnecessary duplicate copies
local_dir_use_symlinks="auto",
ignore_patterns=args.ignore_patterns, ignore_patterns=args.ignore_patterns,
token=args.hf_token, token=args.hf_token,
library_name="llama-toolchain",
) )
except GatedRepoError: except GatedRepoError:
self.parser.error( self.parser.error(