Added Classification from baseinvocation

This commit is contained in:
Jonathan 2023-12-13 18:46:24 -06:00 committed by psychedelicious
parent 5ad88c7f86
commit 18b2bcbbee

View File

@ -13,7 +13,7 @@ from invokeai.app.shared.fields import FieldDescriptions
from invokeai.backend.image_util.invisible_watermark import InvisibleWatermark
from invokeai.backend.image_util.safety_checker import SafetyChecker
from .baseinvocation import BaseInvocation, Input, InputField, InvocationContext, WithMetadata, invocation
from .baseinvocation import BaseInvocation, Classification, Input, InputField, InvocationContext, WithMetadata, invocation
@invocation("show_image", title="Show Image", tags=["image"], category="image", version="1.0.0")