fix cogvideox autoencoder decode (huggingface#9569)

Co-authored-by: Aryan <[email protected]>
tedmeftah · Oct 2, 2024 · 7f323f0 · 7f323f0
1 parent 61d3764
commit 7f323f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py b/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py
@@ -1234,7 +1234,7 @@ def _decode(self, z: torch.Tensor, return_dict: bool = True) -> Union[DecoderOut
             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 = []