add download command

2023-11-16 21:12:06 +08:00
parent 20e660aa4a
commit 1d145d1cd6
17 changed files with 233 additions and 67 deletions
--- a/lama_cleaner/model/manga.py
+++ b/lama_cleaner/model/manga.py
@@ -7,7 +7,7 @@ import torch
 import time
 from loguru import logger

-from lama_cleaner.helper import get_cache_path_by_url, load_jit_model
+from lama_cleaner.helper import get_cache_path_by_url, load_jit_model, download_model
 from lama_cleaner.model.base import InpaintModel
 from lama_cleaner.schema import Config

@@ -42,6 +42,11 @@ class Manga(InpaintModel):
        )
        self.seed = 42

+    @staticmethod
+    def download():
+        download_model(MANGA_INPAINTOR_MODEL_URL, MANGA_INPAINTOR_MODEL_MD5)
+        download_model(MANGA_LINE_MODEL_URL, MANGA_LINE_MODEL_MD5)
+
    @staticmethod
    def is_downloaded() -> bool:
        model_paths = [