phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

download inside model_name directory

Browse source
This commit is contained in:
Hardik Shah 2024-07-20 23:16:10 -07:00
parent 0746a0f62b
commit 2428701951
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
toolchain/cli/download.py
View file

@ -63,9 +63,11 @@ class Download(Subcommand):
os.makedirs(DEFAULT_OUTPUT_DIR, exist_ok=True) os.makedirs(DEFAULT_OUTPUT_DIR, exist_ok=True)
output_dir = args.output_dir output_dir = args.output_dir
model_name = args.repo_id.split("/")[-1]
if output_dir is None: if output_dir is None:
model_name = args.repo_id.split("/")[-1]
output_dir = Path(DEFAULT_OUTPUT_DIR) / model_name output_dir = Path(DEFAULT_OUTPUT_DIR) / model_name
else:
output_dir = Path(output_dir) / model_name
try: try:
true_output_dir = snapshot_download( true_output_dir = snapshot_download(