AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

make style

Browse Source
This commit is contained in:
Daniel Gu
2023-05-11 03:28:37 -07:00
parent 19a20a55b9
commit 28dda62f53
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py
View File

@@ -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