From 9ec46d718d896ba63deab2a91646442a7ac9cba1 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Wed, 7 Aug 2024 16:17:51 -0700
Subject: [PATCH] Show message about checksum file so users can check
 themselves

---
 llama_toolchain/cli/download.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/llama_toolchain/cli/download.py b/llama_toolchain/cli/download.py
index 02cbe40a1..1b6298491 100644
--- a/llama_toolchain/cli/download.py
+++ b/llama_toolchain/cli/download.py
@@ -132,6 +132,9 @@ def _meta_download(model: "Model", meta_url: str):
         downloader = ResumableDownloader(url, output_file, total_size)
         asyncio.run(downloader.download())
 
+    print(f"\nSuccessfully downloaded model to {output_dir}")
+    cprint(f"\nMD5 Checksums are at: {output_dir / 'checklist.chk'}", "white")
+
 
 def run_download_cmd(args: argparse.Namespace, parser: argparse.ArgumentParser):
     from llama_models.sku_list import resolve_model