runner

runner/inference.py

@@ -214,14 +214,15 @@ def download_infercence_cache(configs: Any, model_version: str = "v0.2.0") -> No
             tos_url = URL[cache_name]
             logger.info(f"Downloading data cache from\n {tos_url}...")
             urllib.request.urlretrieve(tos_url, cache_path)
-
-    # Google Drive file ID
-    file_id = '17zBIRed3xZM8ux0bq2hpf1oFC75Y7OEw'
-    # URL to download the file
-    url = f'https://drive.google.com/uc?id={file_id}'
-
-    # Download the file and save it as 'checkpoint.pt'
-    gdown.download(url, './checkpoint.pt', quiet=False)
+            
+    if not os.path.exists('./checkpoint.pt'):
+        # Google Drive file ID
+        file_id = '17zBIRed3xZM8ux0bq2hpf1oFC75Y7OEw'
+        # URL to download the file
+        url = f'https://drive.google.com/uc?id={file_id}'
+
+        # Download the file and save it as 'checkpoint.pt'
+        gdown.download(url, './checkpoint.pt', quiet=False)
 
 
     # checkpoint_path = configs.load_checkpoint_path