fix cogvideox autoencoder decode (#9569)

Co-authored-by: Aryan <aryan@huggingface.co>

fix cogvideox autoencoder decode (#9569)
Co-authored-by: Aryan <aryan@huggingface.co>
7f323f0f · Xiangchendong · GitHub · 61d37640 · 7f323f0f
Unverified Commit 7f323f0f authored Oct 03, 2024 by Xiangchendong Committed by GitHub Oct 02, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py ...diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +1 -1

No files found.
--- a/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py
+++ b/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py
@@ -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 = []