From be2c1746d5daaad89f04bb2a4b14688a2a379629 Mon Sep 17 00:00:00 2001 From: Rex Cheng Date: Sun, 18 Aug 2024 23:07:08 +0900 Subject: [PATCH] remove the local option --- cutie/utils/download_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cutie/utils/download_models.py b/cutie/utils/download_models.py index f9350a2..31a956c 100644 --- 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) \ No newline at end of file + download_models_if_needed() \ No newline at end of file