Merge pull request #20 from jhj0517/feature/update-docs

Update default model to sam2
jhj0517 · Oct 21, 2024 · d1f3027 · d1f3027
2 parents 1783559 + cb3ad9a
commit d1f3027
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/model_downloader.py b/modules/model_downloader.py
@@ -3,7 +3,7 @@
 
 from modules.paths import *
 
-DEFAULT_MODEL_TYPE = "sam2.1_hiera_large"
+DEFAULT_MODEL_TYPE = "sam2_hiera_large"
 AVAILABLE_MODELS = {
     "sam2_hiera_tiny": ["sam2_hiera_tiny.pt", "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt"],
     "sam2_hiera_small": ["sam2_hiera_small.pt", "https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_small.pt"],