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/scripts/amg.py
+++ b/scripts/amg.py
@@ -41,8 +41,8 @@ parser.add_argument(
 parser.add_argument(
    "--model-type",
    type=str,
-    default="default",
-    help="The type of model to load, in ['default', 'vit_l', 'vit_b']",
+    required=True,
+    help="The type of model to load, in ['default', 'vit_h', 'vit_l', 'vit_b']",
 )

 parser.add_argument(