add MAT model

2022-08-22 23:24:02 +08:00
parent a5e840765e
commit 6d2b24ed6b
8 changed files with 2132 additions and 9 deletions
--- a/lama_cleaner/model_manager.py
+++ b/lama_cleaner/model_manager.py
@@ -1,12 +1,14 @@
 from lama_cleaner.model.lama import LaMa
 from lama_cleaner.model.ldm import LDM
+from lama_cleaner.model.mat import MAT
 from lama_cleaner.model.zits import ZITS
 from lama_cleaner.schema import Config

 models = {
    'lama': LaMa,
    'ldm': LDM,
-    'zits': ZITS
+    'zits': ZITS,
+    'mat': MAT
 }