From a537b8b795f1dd9d43a7302d46dd36b3ab82e023 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Fri, 30 Aug 2024 17:15:11 -0700
Subject: [PATCH] add utils to download files

---
 spaces.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/spaces.py b/spaces.py
index d592722d..05370f06 100644
--- a/spaces.py
+++ b/spaces.py
@@ -180,6 +180,29 @@ def convert_root_path():
     return result + '/'
 
 
+def download_single_file(
+    url: str,
+    *,
+    model_dir: str,
+    progress: bool = True,
+    file_name: str | None = None,
+    hash_prefix: str | None = None,
+) -> str:
+    os.makedirs(model_dir, exist_ok=True)
+    if not file_name:
+        from urllib.parse import urlparse
+        parts = urlparse(url)
+        file_name = os.path.basename(parts.path)
+    cached_file = os.path.abspath(os.path.join(model_dir, file_name))
+    if not os.path.exists(cached_file):
+        tmp_filename = cached_file + '.tmp'
+        print(f'Downloading: "{url}" to {cached_file} using temp file {tmp_filename}\n')
+        from torch.hub import download_url_to_file
+        download_url_to_file(url, tmp_filename, progress=progress, hash_prefix=hash_prefix)
+        os.replace(tmp_filename, cached_file)
+    return cached_file
+
+
 def automatically_move_to_gpu_when_forward(m: torch.nn.Module, target_model: torch.nn.Module = None):
     if target_model is None:
         target_model = m