Skip to content

Commit c3a6a6f

Browse files
committed
Rename SegmentAnythingModelInvocation -> SegmentAnythingInvocation.
1 parent b9dc346 commit c3a6a6f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

invokeai/app/invocations/segment_anything_model.py renamed to invokeai/app/invocations/segment_anything.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,13 @@
1919

2020

2121
@invocation(
22-
"segment_anything_model",
23-
title="Segment Anything Model",
22+
"segment_anything",
23+
title="Segment Anything",
2424
tags=["prompt", "segmentation"],
2525
category="segmentation",
2626
version="1.0.0",
2727
)
28-
class SegmentAnythingModelInvocation(BaseInvocation):
28+
class SegmentAnythingInvocation(BaseInvocation):
2929
"""Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643).
3030
3131
Reference:
@@ -88,7 +88,7 @@ def _segment(
8888

8989
with (
9090
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
9292
) as sam_pipeline,
9393
):
9494
assert isinstance(sam_pipeline, SegmentAnythingPipeline)

0 commit comments

Comments
 (0)