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

Fixes

Browse Source
This commit is contained in:
Sergey Borisov
2023-06-12 16:14:09 +03:00
parent 9fa78443de
commit 36eb1bd893
5 changed files with 71 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -64,7 +64,7 @@ class ModelLoaderInvocation(BaseInvocation):
def invoke(self, context: InvocationContext) -> ModelLoaderOutput:
base_model = BaseModelType.StableDiffusion2 # TODO:
base_model = BaseModelType.StableDiffusion1_5 # TODO:
# TODO: not found exceptions
if not context.services.model_manager.model_exists(
@ -116,7 +116,7 @@ class ModelLoaderInvocation(BaseInvocation):
model_name=self.model_name,
base_model=base_model,
model_type=ModelType.Pipeline,
submodel=SDModelType.Scheduler,
submodel=SubModelType.Scheduler,
),
loras=[],
),
@ -125,13 +125,13 @@ class ModelLoaderInvocation(BaseInvocation):
model_name=self.model_name,
base_model=base_model,
model_type=ModelType.Pipeline,
submodel=SDModelType.Tokenizer,
submodel=SubModelType.Tokenizer,
),
text_encoder=ModelInfo(
model_name=self.model_name,
base_model=base_model,
model_type=ModelType.Pipeline,
submodel=SDModelType.TextEncoder,
submodel=SubModelType.TextEncoder,
),
loras=[],
),
@ -140,7 +140,7 @@ class ModelLoaderInvocation(BaseInvocation):
model_name=self.model_name,
base_model=base_model,
model_type=ModelType.Pipeline,
submodel=SDModelType.Vae,
submodel=SubModelType.Vae,
),
)
)