fix: ✨ handle non fork gdown in model dll

Also fix a dumb mistake from earlier tests..
melMass · Jul 7, 2023 · 95797e8 · 95797e8
1 parent 1e28606
commit 95797e8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/scripts/download_models.py b/scripts/download_models.py
@@ -43,7 +43,7 @@
         "download_url": [
             "https://drive.google.com/drive/folders/131_--QrieM4aQbbLWrUtbO2cGbX8-war"
         ],
-        "destination": "FILMOS",
+        "destination": "FILM",
     },
 }
 
@@ -65,7 +65,11 @@ def download_model(download_url, destination):
     if "drive.google.com" in download_url:
         if "/folders/" in download_url:
             # download folder
-            gdown.download_folder(download_url, output=destination, resume=True)
+            try:
+                gdown.download_folder(download_url, output=destination, resume=True)
+            except TypeError:
+                gdown.download_folder(download_url, output=destination)
+
             return
         # download from google drive
         gdown.download(download_url, destination, quiet=False, resume=True)