remove the local option

hkchengrex · Aug 18, 2024 · be2c174 · be2c174
1 parent b2dc761
commit be2c174
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cutie/utils/download_models.py b/cutie/utils/download_models.py
@@ -10,7 +10,7 @@
     ('https://github.com/hkchengrex/Cutie/releases/download/v1.0/cutie-base-mega.pth', 'a6071de6136982e396851903ab4c083a'),
 ]
 
-def download_models_if_needed(local: bool = False) -> str:
+def download_models_if_needed() -> str:
     weight_dir = os.path.join(os.path.dirname(__file__), '..', '..', 'weights')
     os.makedirs(weight_dir, exist_ok=True)
     for link, md5 in _links:
@@ -33,4 +33,4 @@ def download_models_if_needed(local: bool = False) -> str:
 
 
 if __name__ == '__main__':
-    download_models_if_needed(local=True)
+    download_models_if_needed()