update

2023-12-24 15:32:27 +08:00
parent 0e5e16ba20
commit 371db2d771
31 changed files with 441 additions and 439 deletions
--- a/lama_cleaner/plugins/segment_anything/build_sam.py
+++ b/lama_cleaner/plugins/segment_anything/build_sam.py
@@ -109,7 +109,7 @@ sam_model_registry = {
    "vit_h": build_sam,
    "vit_l": build_sam_vit_l,
    "vit_b": build_sam_vit_b,
-    "vit_t": build_sam_vit_t,
+    "mobile_sam": build_sam_vit_t,
 }