remove direct call to build_sam()

This commit is contained in:
Hanzi Mao committed 2023-04-07 15:18:56 -07:00
1 parent 7c524018a6
commit 8991c14de4
4 files changed
+13 -18

No files matched your search

+2 -2
View File
@@ -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,
}