fix

2023-05-12 17:28:35 +08:00
parent ca470db570
commit 0363472adc
2 changed files with 2 additions and 2 deletions
--- a/lama_cleaner/tests/test_model_md5.py
+++ b/lama_cleaner/tests/test_model_md5.py
@@ -2,7 +2,7 @@ def test_load_model():
    from lama_cleaner.plugins import InteractiveSeg
    from lama_cleaner.model_manager import ModelManager

-    interactive_seg_model = InteractiveSeg()
+    interactive_seg_model = InteractiveSeg('vit_l', 'cpu')

    models = [
        "lama",