Spaces:
Runtime error
Runtime error
jhj0517
commited on
Commit
•
c5a4e30
1
Parent(s):
0f36b51
Chore: Update variable name
Browse files- modules/sam_inference.py +2 -2
modules/sam_inference.py
CHANGED
@@ -21,7 +21,7 @@ from modules.mask_utils import (
|
|
21 |
)
|
22 |
from modules.logger_util import get_logger
|
23 |
|
24 |
-
|
25 |
"sam2_hiera_tiny": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_t.yaml"),
|
26 |
"sam2_hiera_small": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_s.yaml"),
|
27 |
"sam2_hiera_base_plus": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_b+.yaml"),
|
@@ -48,7 +48,7 @@ class SamInference:
|
|
48 |
|
49 |
def load_model(self,
|
50 |
load_video_predictor: bool = False):
|
51 |
-
config =
|
52 |
filename, url = AVAILABLE_MODELS[self.model_type]
|
53 |
model_path = os.path.join(self.model_dir, filename)
|
54 |
|
|
|
21 |
)
|
22 |
from modules.logger_util import get_logger
|
23 |
|
24 |
+
MODEL_CONFIGS = {
|
25 |
"sam2_hiera_tiny": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_t.yaml"),
|
26 |
"sam2_hiera_small": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_s.yaml"),
|
27 |
"sam2_hiera_base_plus": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_b+.yaml"),
|
|
|
48 |
|
49 |
def load_model(self,
|
50 |
load_video_predictor: bool = False):
|
51 |
+
config = MODEL_CONFIGS[self.model_type]
|
52 |
filename, url = AVAILABLE_MODELS[self.model_type]
|
53 |
model_path = os.path.join(self.model_dir, filename)
|
54 |
|