Skip to content

Commit 7a91ea6

Browse files
authored
Remove remaining not in upscale pipeline (#4020)
Remove remaining `not` in upscale pipeline.
1 parent e4559f4 commit 7a91ea6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -760,7 +760,7 @@ def __call__(
760760

761761
# if xformers or torch_2_0 is used attention block does not need
762762
# to be in float32 which can save lots of memory
763-
if not use_torch_2_0_or_xformers:
763+
if use_torch_2_0_or_xformers:
764764
self.vae.post_quant_conv.to(latents.dtype)
765765
self.vae.decoder.conv_in.to(latents.dtype)
766766
self.vae.decoder.mid_block.to(latents.dtype)

0 commit comments

Comments
 (0)