From 28dda62f533fe23afc44074defecc3c1eefca6d8 Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Thu, 11 May 2023 03:28:37 -0700 Subject: [PATCH] make style --- .../pipelines/unidiffuser/pipeline_unidiffuser.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py b/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py index 30219ca6c2..d5f598c311 100644 --- a/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py +++ b/src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py @@ -771,9 +771,7 @@ class UniDiffuserPipeline(DiffusionPipeline): img_vae, img_clip = x.split([img_vae_dim, self.image_encoder_hidden_size], dim=1) - img_vae = torch.reshape( - img_vae, (batch_size, self.num_channels_latents, latent_height, latent_width) - ) + img_vae = torch.reshape(img_vae, (batch_size, self.num_channels_latents, latent_height, latent_width)) img_clip = torch.reshape(img_clip, (batch_size, 1, self.image_encoder_hidden_size)) return img_vae, img_clip @@ -800,9 +798,7 @@ class UniDiffuserPipeline(DiffusionPipeline): img_vae, img_clip, text = x.split([img_vae_dim, self.image_encoder_hidden_size, text_dim], dim=1) - img_vae = torch.reshape( - img_vae, (batch_size, self.num_channels_latents, latent_height, latent_width) - ) + img_vae = torch.reshape(img_vae, (batch_size, self.num_channels_latents, latent_height, latent_width)) img_clip = torch.reshape(img_clip, (batch_size, 1, self.image_encoder_hidden_size)) text = torch.reshape(text, (batch_size, self.text_encoder_seq_len, self.text_encoder_hidden_size)) return img_vae, img_clip, text