Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename SegmentAnythingModelInvocation -> SegmentAnythingInvocation.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-01 10:00:36 -04:00
parent b9dc3460ba
commit c3a6a6fb22
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/segment_anything_model.py → invokeai/app/invocations/segment_anything.py
View File

@ -19,13 +19,13 @@ SEGMENT_ANYTHING_MODEL_ID = "facebook/sam-vit-base"
@invocation( @invocation(
"segment_anything_model", "segment_anything",
title="Segment Anything Model", title="Segment Anything",
tags=["prompt", "segmentation"], tags=["prompt", "segmentation"],
category="segmentation", category="segmentation",
version="1.0.0", version="1.0.0",
) )
class SegmentAnythingModelInvocation(BaseInvocation): class SegmentAnythingInvocation(BaseInvocation):
"""Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643). """Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643).
Reference: Reference:
@ -88,7 +88,7 @@ class SegmentAnythingModelInvocation(BaseInvocation):
with ( with (
context.models.load_remote_model( 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, ) as sam_pipeline,
): ):
assert isinstance(sam_pipeline, SegmentAnythingPipeline) assert isinstance(sam_pipeline, SegmentAnythingPipeline)