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

VAE loader is loading proper VAE. Unclear if it is changing the image

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-30 18:46:36 -04:00 committed by psychedelicious
parent fa8a5838d3
commit a8a2209560
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -254,9 +254,9 @@ class VaeLoaderInvocation(BaseInvocation):
}
def invoke(self, context: InvocationContext) -> VaeLoaderOutput:
base_model = self.vae.base_model
model_name = self.vae.model_name
model_type = ModelType.vae
base_model = self.vae_model.base_model
model_name = self.vae_model.model_name
model_type = ModelType.Vae
if not context.services.model_manager.model_exists(
base_model=base_model,
@ -266,8 +266,10 @@ class VaeLoaderInvocation(BaseInvocation):
raise Exception(f"Unkown vae name: {model_name}!")
return VaeLoaderOutput(
vae=VaeField(
vae = ModelInfo(
model_name = model_name,
base_model = base_model,
model_type = model_type,
)
)
)