kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Fix bugs] pipeline_controlnet_sd_xl.py (#6653)

Browse Source 
* Update pipeline_controlnet_sd_xl.py

* Update pipeline_controlnet_xs_sd_xl.py
This commit is contained in:
Haofan Wang
2024-01-24 11:48:12 +08:00
committed by GitHub
parent 0eb68d9ddb
commit c9081a8abd
2 changed files with 0 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/research_projects/controlnetxs/pipeline_controlnet_xs_sd_xl.py
-5
View File
@@ -1041,11 +1041,6 @@ class StableDiffusionXLControlNetXSPipeline(
step_idx = i // getattr(self.scheduler, "order", 1)
callback(step_idx, t, latents)
# manually for max memory savings
if self.vae.dtype == torch.float16 and self.vae.config.force_upcast:
self.upcast_vae()
latents = latents.to(next(iter(self.vae.post_quant_conv.parameters())).dtype)
if not output_type == "latent":
# make sure the VAE is in float32 mode, as it overflows in float16
needs_upcasting = self.vae.dtype == torch.float16 and self.vae.config.force_upcast
src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py
-5
View File
@@ -1404,11 +1404,6 @@ class StableDiffusionXLControlNetPipeline(
step_idx = i // getattr(self.scheduler, "order", 1)
callback(step_idx, t, latents)
# manually for max memory savings
if self.vae.dtype == torch.float16 and self.vae.config.force_upcast:
self.upcast_vae()
latents = latents.to(next(iter(self.vae.post_quant_conv.parameters())).dtype)
if not output_type == "latent":
# make sure the VAE is in float32 mode, as it overflows in float16
needs_upcasting = self.vae.dtype == torch.float16 and self.vae.config.force_upcast