From 24287019519a4bdc8dee20af8ccb82467126c040 Mon Sep 17 00:00:00 2001
From: Hardik Shah <hjshah@fb.com>
Date: Sat, 20 Jul 2024 23:16:10 -0700
Subject: [PATCH] download inside model_name directory

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

diff --git a/toolchain/cli/download.py b/toolchain/cli/download.py
index edb1eb3a3..a905cca55 100644
--- a/toolchain/cli/download.py
+++ b/toolchain/cli/download.py
@@ -63,9 +63,11 @@ class Download(Subcommand):
 
         os.makedirs(DEFAULT_OUTPUT_DIR, exist_ok=True)
         output_dir = args.output_dir
+        model_name = args.repo_id.split("/")[-1]
         if output_dir is None:
-            model_name = args.repo_id.split("/")[-1]
             output_dir = Path(DEFAULT_OUTPUT_DIR) / model_name
+        else:
+            output_dir = Path(output_dir) / model_name
 
         try:
             true_output_dir = snapshot_download(