wip

2023-12-01 10:15:35 +08:00
parent 973987dfbb
commit 9a9eb8abfd
55 changed files with 2596 additions and 1251 deletions
--- a/lama_cleaner/model/lama.py
+++ b/lama_cleaner/model/lama.py
@@ -23,6 +23,7 @@ LAMA_MODEL_MD5 = os.environ.get("LAMA_MODEL_MD5", "e3aa4aaa15225a33ec84f9f4bc47e
 class LaMa(InpaintModel):
    name = "lama"
    pad_mod = 8
+    is_erase_model = True

    @staticmethod
    def download():