modif workflows loader

roboflow · Aug 15, 2024 · 1a9c73b · 1a9c73b
1 parent 68b3e34
commit 1a9c73b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/inference/core/workflows/core_steps/loader.py b/inference/core/workflows/core_steps/loader.py
@@ -44,6 +44,9 @@
 from inference.core.workflows.core_steps.models.foundation.yolo_world.v1 import (
     YoloWorldModelBlockV1,
 )
+from inference.core.workflows.core_steps.models.foundation.segment_anything2.v1 import (
+    SegmentAnything2BlockV1,
+)
 from inference.core.workflows.core_steps.models.roboflow.instance_segmentation.v1 import (
     RoboflowInstanceSegmentationModelBlockV1,
 )
@@ -209,6 +212,7 @@
 
 def load_blocks() -> List[Type[WorkflowBlock]]:
     return [
+        SegmentAnything2BlockV1,
         DetectionsConsensusBlockV1,
         ClipComparisonBlockV1,
         LMMBlockV1,