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/zits.py
+++ b/lama_cleaner/model/zits.py
@@ -216,7 +216,7 @@ class ZITS(InpaintModel):
        self.device = device
        self.sample_edge_line_iterations = 1

-    def init_model(self, device):
+    def init_model(self, device, **kwargs):
        self.wireframe = load_jit_model(ZITS_WIRE_FRAME_MODEL_URL, device)
        self.edge_line = load_jit_model(ZITS_EDGE_LINE_MODEL_URL, device)
        self.structure_upsample = load_jit_model(