load the correct model type

pytorch · Feb 8, 2025 · f147fdc · f147fdc
1 parent fa95dcf
commit f147fdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/sam2_amg_server/cli_on_modal.py b/examples/sam2_amg_server/cli_on_modal.py
@@ -120,7 +120,7 @@ def build(self):
         )
         from compile_export_utils import load_exported_model
         export_model_path = Path(TARGET) / Path("exported_models")
-        export_model_path = export_model_path / Path("sam2") / Path("sam2_amg")
+        export_model_path = export_model_path / Path("sam2") / Path(f"sam2_{self.task_type}")
         if not self.baseline:
             load_exported_model(mask_generator,
                                 export_model_path,