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

Found another instance of expecting a mid_block on the decoder in a vae

Browse Source
This commit is contained in:
Brandon Rising 2024-03-12 12:11:38 -04:00
parent 7b393656de
commit 8d2a4db902
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/invocations/latent.py
View File

@ -844,7 +844,6 @@ class LatentsToImageInvocation(BaseInvocation, WithMetadata, WithBoard):
if self.fp32:
vae.to(dtype=torch.float32)
# AutoencoderTiny doesn't contain a mid_block property or appear to accept attn processors
use_torch_2_0_or_xformers = hasattr(vae.decoder, "mid_block") and isinstance(
vae.decoder.mid_block.attentions[0].processor,
(
@ -1019,7 +1018,7 @@ class ImageToLatentsInvocation(BaseInvocation):
if upcast:
vae.to(dtype=torch.float32)
use_torch_2_0_or_xformers = isinstance(
use_torch_2_0_or_xformers = hasattr(vae.decoder, "mid_block") and isinstance(
vae.decoder.mid_block.attentions[0].processor,
(
AttnProcessor2_0,