We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eaf0624 commit 480857aCopy full SHA for 480857a
invokeai/app/invocations/model.py
@@ -153,7 +153,11 @@ def invoke(self, context: InvocationContext) -> ModelIdentifierOutput:
153
class MainModelLoaderInvocation(BaseInvocation):
154
"""Loads a main model, outputting its submodels."""
155
156
- model: ModelIdentifierField = InputField(description=FieldDescriptions.main_model, ui_model_type=ModelType.Main)
+ model: ModelIdentifierField = InputField(
157
+ description=FieldDescriptions.main_model,
158
+ ui_model_base=BaseModelType.StableDiffusion1,
159
+ ui_model_type=ModelType.Main,
160
+ )
161
# TODO: precision?
162
163
def invoke(self, context: InvocationContext) -> ModelLoaderOutput:
0 commit comments