Skip to content

Commit 6af8df9

Browse files
StAlKeR7779hipsterusername
authored andcommitted
Temporary force set vae to same precision as unet
1 parent 8e7eae6 commit 6af8df9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

invokeai/app/invocations/generate.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,8 @@ def _lora_loader():
181181
device = context.services.model_manager.mgr.cache.execution_device
182182
dtype = context.services.model_manager.mgr.cache.precision
183183

184+
vae.to(dtype=unet.dtype)
185+
184186
pipeline = StableDiffusionGeneratorPipeline(
185187
vae=vae,
186188
text_encoder=None,

0 commit comments

Comments
 (0)