From c3a6a6fb22127e4add43e09691ba942032d1d922 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 1 Aug 2024 10:00:36 -0400 Subject: [PATCH] Rename SegmentAnythingModelInvocation -> SegmentAnythingInvocation. --- .../{segment_anything_model.py => segment_anything.py} | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) rename invokeai/app/invocations/{segment_anything_model.py => segment_anything.py} (97%) diff --git a/invokeai/app/invocations/segment_anything_model.py b/invokeai/app/invocations/segment_anything.py similarity index 97% rename from invokeai/app/invocations/segment_anything_model.py rename to invokeai/app/invocations/segment_anything.py index 13fc71f231..5b766f405a 100644 --- a/invokeai/app/invocations/segment_anything_model.py +++ b/invokeai/app/invocations/segment_anything.py @@ -19,13 +19,13 @@ SEGMENT_ANYTHING_MODEL_ID = "facebook/sam-vit-base" @invocation( - "segment_anything_model", - title="Segment Anything Model", + "segment_anything", + title="Segment Anything", tags=["prompt", "segmentation"], category="segmentation", version="1.0.0", ) -class SegmentAnythingModelInvocation(BaseInvocation): +class SegmentAnythingInvocation(BaseInvocation): """Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643). Reference: @@ -88,7 +88,7 @@ class SegmentAnythingModelInvocation(BaseInvocation): with ( context.models.load_remote_model( - source=SEGMENT_ANYTHING_MODEL_ID, loader=SegmentAnythingModelInvocation._load_sam_model + source=SEGMENT_ANYTHING_MODEL_ID, loader=SegmentAnythingInvocation._load_sam_model ) as sam_pipeline, ): assert isinstance(sam_pipeline, SegmentAnythingPipeline)