update

lama_cleaner/tests/test_model_md5.py

@@ -2,16 +2,9 @@ def test_load_model():
     from lama_cleaner.plugins import InteractiveSeg
     from lama_cleaner.model_manager import ModelManager
 
-    interactive_seg_model = InteractiveSeg('vit_l', 'cpu')
+    interactive_seg_model = InteractiveSeg("vit_l", "cpu")
 
-    models = [
-        "lama",
-        "ldm",
-        "zits",
-        "mat",
-        "fcf",
-        "manga",
-    ]
+    models = ["lama", "ldm", "zits", "mat", "fcf", "manga", "migan"]
     for m in models:
         ModelManager(
             name=m,
@@ -21,25 +14,3 @@ def test_load_model():
             sd_cpu_textencoder=True,
             cpu_offload=True,
         )
-
-
-# def create_empty_file(tmp_dir, name):
-#     tmp_model_dir = os.path.join(tmp_dir, "torch", "hub", "checkpoints")
-#     Path(tmp_model_dir).mkdir(exist_ok=True, parents=True)
-#     path = os.path.join(tmp_model_dir, name)
-#     with open(path, "w") as f:
-#         f.write("1")
-#
-#
-# def test_load_model_error():
-#     MODELS = [
-#         ("big-lama.pt", "e3aa4aaa15225a33ec84f9f4bc47e500"),
-#         ("cond_stage_model_encode.pt", "23239fc9081956a3e70de56472b3f296"),
-#         ("cond_stage_model_decode.pt", "fe419cd15a750d37a4733589d0d3585c"),
-#         ("diffusion.pt", "b0afda12bf790c03aba2a7431f11d22d"),
-#     ]
-#     with tempfile.TemporaryDirectory() as tmp_dir:
-#         os.environ["XDG_CACHE_HOME"] = tmp_dir
-#         for name, md5 in MODELS:
-#             create_empty_file(tmp_dir, name)
-#             test_load_model()