rename to iopaint

2024-01-05 15:19:23 +08:00
parent f1f18aa6cd
commit a73e2a531f
101 changed files with 180 additions and 253 deletions
--- a/iopaint/tests/test_model_md5.py
+++ b/iopaint/tests/test_model_md5.py
@@ -0,0 +1,16 @@
+def test_load_model():
+    from iopaint.plugins import InteractiveSeg
+    from iopaint.model_manager import ModelManager
+
+    interactive_seg_model = InteractiveSeg("vit_l", "cpu")
+
+    models = ["lama", "ldm", "zits", "mat", "fcf", "manga", "migan"]
+    for m in models:
+        ModelManager(
+            name=m,
+            device="cpu",
+            no_half=False,
+            disable_nsfw=False,
+            sd_cpu_textencoder=True,
+            cpu_offload=True,
+        )