Delete models from old location for huggingface download

pytorch · Oct 9, 2024 · 0ca3941 · 0ca3941
1 parent a4aaccf
commit 0ca3941
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torchchat/cli/download.py b/torchchat/cli/download.py
@@ -149,6 +149,12 @@ def download_and_convert(
         model_config.distribution_channel
         == ModelDistributionChannel.HuggingFaceSnapshot
     ):
+        # Check if model is already downloaded in the old location
+
+        if os.path.exists(models_dir / model_config.name):
+            print(f"Found model artifacts in {models_dir / model_config.name}. Downloading in Hugging Face cache...")
+            shutil.rmtree(models_dir / model_config.name)
+
         _download_hf_snapshot(model_config, hf_token)
         return