Skip to content

Commit 480857a

Browse files
fix(nodes): add base to SD1 model loader
1 parent eaf0624 commit 480857a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

invokeai/app/invocations/model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,11 @@ def invoke(self, context: InvocationContext) -> ModelIdentifierOutput:
153153
class MainModelLoaderInvocation(BaseInvocation):
154154
"""Loads a main model, outputting its submodels."""
155155

156-
model: ModelIdentifierField = InputField(description=FieldDescriptions.main_model, ui_model_type=ModelType.Main)
156+
model: ModelIdentifierField = InputField(
157+
description=FieldDescriptions.main_model,
158+
ui_model_base=BaseModelType.StableDiffusion1,
159+
ui_model_type=ModelType.Main,
160+
)
157161
# TODO: precision?
158162

159163
def invoke(self, context: InvocationContext) -> ModelLoaderOutput:

0 commit comments

Comments
 (0)