wip

2022-09-15 22:21:27 +08:00
parent 3ac6ee7f44
commit 32854d40da
52 changed files with 2258 additions and 205 deletions
--- a/lama_cleaner/model/mat.py
+++ b/lama_cleaner/model/mat.py
@@ -1405,7 +1405,7 @@ class MAT(InpaintModel):
    pad_mod = 512
    pad_to_square = True

-    def init_model(self, device):
+    def init_model(self, device, **kwargs):
        seed = 240  # pick up a random number
        random.seed(seed)
        np.random.seed(seed)