make model switch work with toast

2022-04-17 23:31:12 +08:00
parent 205286a414
commit f7e1e073dc
18 changed files with 447 additions and 28 deletions
--- a/lama_cleaner/model/base.py
+++ b/lama_cleaner/model/base.py
@@ -24,6 +24,11 @@ class InpaintModel:
    def init_model(self, device):
        ...

+    @staticmethod
+    @abc.abstractmethod
+    def is_downloaded() -> bool:
+        ...
+
    @abc.abstractmethod
    def forward(self, image, mask, config: Config):
        """Input image and output image have same size