From 7f8dc9ba677643d0477dd39f3c27edb1844cce98 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 30 Jul 2024 22:05:31 +0300
Subject: [PATCH] Update load_model.py

---
 load_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/load_model.py b/load_model.py
index 95d343b..93e6673 100644
--- a/load_model.py
+++ b/load_model.py
@@ -185,5 +185,7 @@ def load_model(model_path, model_cfg_path, segmentor, dtype, device):
         model_class = SAM2VideoPredictor
         model = initialize_model(model_class, model_config, segmentor, image_encoder, memory_attention, memory_encoder, sam_mask_decoder_extra_args, dtype, device)
         model.load_state_dict(sd)
+    else:
+        raise ValueError(f"Segmentor {segmentor} not supported")
 
     return model
\ No newline at end of file