diff --git a/llama_stack/cli/download.py b/llama_stack/cli/download.py index bb57186e5..c2f8ac855 100644 --- a/llama_stack/cli/download.py +++ b/llama_stack/cli/download.py @@ -380,6 +380,7 @@ def _hf_download( def _meta_download( model: "Model", + model_id: str, meta_url: str, info: "LlamaDownloadInfo", max_concurrent_downloads: int, @@ -405,8 +406,15 @@ def _meta_download( downloader = ParallelDownloader(max_concurrent_downloads=max_concurrent_downloads) asyncio.run(downloader.download_all(tasks)) - print(f"\nSuccessfully downloaded model to {output_dir}") - cprint(f"\nMD5 Checksums are at: {output_dir / 'checklist.chk'}", "white") + cprint(f"\nSuccessfully downloaded model to {output_dir}", "green") + cprint( + f"\nView MD5 checksum files at: {output_dir / 'checklist.chk'}", + "white", + ) + cprint( + f"\n[Optionally] To run MD5 checksums, use the following command: llama model verify-download --model-id {model_id}", + "yellow", + ) class ModelEntry(BaseModel): @@ -512,7 +520,7 @@ def run_download_cmd(args: argparse.Namespace, parser: argparse.ArgumentParser): ) if "llamameta.net" not in meta_url: parser.error("Invalid Meta URL provided") - _meta_download(model, meta_url, info, args.max_parallel) + _meta_download(model, model_id, meta_url, info, args.max_parallel) except Exception as e: parser.error(f"Download failed: {str(e)}")