Fix bug with VAE slicing in autoencoder_dc.py (#12343)

d83d35c1 · Chen Mingyi · GitHub · 843355f8 · d83d35c1
Unverified Commit d83d35c1 authored Sep 22, 2025 by Chen Mingyi Committed by GitHub Sep 22, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/src/diffusers/models/autoencoders/autoencoder_dc.py
+++ b/src/diffusers/models/autoencoders/autoencoder_dc.py
@@ -617,7 +617,7 @@ class AutoencoderDC(ModelMixin, ConfigMixin, FromOriginalModelMixin):
                returned.
        """
        if self.use_slicing and z.size(0) > 1:
-            decoded_slices = [self._decode(z_slice).sample for z_slice in z.split(1)]
+            decoded_slices = [self._decode(z_slice) for z_slice in z.split(1)]
            decoded = torch.cat(decoded_slices)
        else:
            decoded = self._decode(z)