From 8ca910535efd045cc84c679005ad77012316e0ee Mon Sep 17 00:00:00 2001 From: Xiang-cd Date: Wed, 2 Oct 2024 14:49:56 +0800 Subject: [PATCH] fix cogvideox autoencoder decode --- src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py b/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py index 04c787ee3e84..6acb73903cad 100644 --- a/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py +++ b/src/diffusers/models/autoencoders/autoencoder_kl_cogvideox.py @@ -1142,7 +1142,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) dec = [] for i in range(num_batches): remaining_frames = num_frames % frame_batch_size