File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 19
19
20
20
21
21
@invocation (
22
- "segment_anything_model " ,
23
- title = "Segment Anything Model " ,
22
+ "segment_anything " ,
23
+ title = "Segment Anything" ,
24
24
tags = ["prompt" , "segmentation" ],
25
25
category = "segmentation" ,
26
26
version = "1.0.0" ,
27
27
)
28
- class SegmentAnythingModelInvocation (BaseInvocation ):
28
+ class SegmentAnythingInvocation (BaseInvocation ):
29
29
"""Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643).
30
30
31
31
Reference:
@@ -88,7 +88,7 @@ def _segment(
88
88
89
89
with (
90
90
context .models .load_remote_model (
91
- source = SEGMENT_ANYTHING_MODEL_ID , loader = SegmentAnythingModelInvocation ._load_sam_model
91
+ source = SEGMENT_ANYTHING_MODEL_ID , loader = SegmentAnythingInvocation ._load_sam_model
92
92
) as sam_pipeline ,
93
93
):
94
94
assert isinstance (sam_pipeline , SegmentAnythingPipeline )
You can’t perform that action at this time.
0 commit comments