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

feat(nodes): make ModelIdentifierInvocation a prototype

Browse Source
This commit is contained in:
psychedelicious 2024-05-19 19:26:17 +10:00
parent cea1874e00
commit 281bd31db2
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/invocations/model.py
View File

@ -11,6 +11,7 @@ from invokeai.backend.model_manager.config import AnyModelConfig, BaseModelType,
from .baseinvocation import ( from .baseinvocation import (
BaseInvocation, BaseInvocation,
BaseInvocationOutput, BaseInvocationOutput,
Classification,
invocation, invocation,
invocation_output, invocation_output,
) )
@ -106,9 +107,12 @@ class ModelIdentifierOutput(BaseInvocationOutput):
tags=["model"], tags=["model"],
category="model", category="model",
version="1.0.0", version="1.0.0",
classification=Classification.Prototype,
) )
class ModelIdentifierInvocation(BaseInvocation): class ModelIdentifierInvocation(BaseInvocation):
"""Selects any model, outputting it.""" """Selects any model, outputting it its identifier. Be careful with this one! The identifier will be accepted as
input for any model, even if the model types don't match. If you connect this to a mismatched input, you'll get an
error."""
model: ModelIdentifierField = InputField(description="The model to select", title="Model") model: ModelIdentifierField = InputField(description="The model to select", title="Model")