fix model init

2022-09-22 22:45:24 +08:00
parent 8d65195e8a
commit 5bdc5c1526
2 changed files with 18 additions and 13 deletions
--- a/lama_cleaner/model/zits.py
+++ b/lama_cleaner/model/zits.py
@@ -206,7 +206,7 @@ class ZITS(InpaintModel):
    pad_mod = 32
    pad_to_square = True

-    def __init__(self, device):
+    def __init__(self, device, **kwargs):
        """

        Args: