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

fix cogvideox autoencoder decode (#9569)

Browse Source 
Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
Xiangchendong
2024-10-03 03:07:06 +08:00
committed by GitHub
parent 61d37640ad
commit 7f323f0f31
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py
View File

@@ -1234,7 +1234,7 @@ class AutoencoderKLCogVideoX(ModelMixin, ConfigMixin, FromOriginalModelMixin):
return self.tiled_decode(z, return_dict=return_dict)
frame_batch_size = self.num_latent_frames_batch_size
num_batches = num_frames // frame_batch_size
num_batches = max(num_frames // frame_batch_size, 1)
conv_cache = None
dec = []