"tests/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "ff5b5a4a1090f79e82163f43909a1831b7fce924"
Unverified Commit 7a91ea6c authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Remove remaining `not` in upscale pipeline (#4020)

Remove remaining `not` in upscale pipeline.
parent e4559f48
...@@ -760,7 +760,7 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversionLoaderMi ...@@ -760,7 +760,7 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversionLoaderMi
# if xformers or torch_2_0 is used attention block does not need # if xformers or torch_2_0 is used attention block does not need
# to be in float32 which can save lots of memory # to be in float32 which can save lots of memory
if not use_torch_2_0_or_xformers: if use_torch_2_0_or_xformers:
self.vae.post_quant_conv.to(latents.dtype) self.vae.post_quant_conv.to(latents.dtype)
self.vae.decoder.conv_in.to(latents.dtype) self.vae.decoder.conv_in.to(latents.dtype)
self.vae.decoder.mid_block.to(latents.dtype) self.vae.decoder.mid_block.to(latents.dtype)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment