remove direct call to build_sam()

2023-04-07 15:13:18 -07:00
parent 7c524018a6
commit 8991c14de4
4 changed files with 13 additions and 18 deletions
--- a/segment_anything/build_sam.py
+++ b/segment_anything/build_sam.py
@@ -45,8 +45,8 @@ def build_sam_vit_b(checkpoint=None):


 sam_model_registry = {
-    "default": build_sam,
-    "vit_h": build_sam,
+    "default": build_sam_vit_h,
+    "vit_h": build_sam_vit_h,
    "vit_l": build_sam_vit_l,
    "vit_b": build_sam_vit_b,
 }