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

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

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