Make VAE compatible to torch.compile() (huggingface#7984)

make VAE compatible to torch.compile()

Co-authored-by: YiYi Xu <[email protected]>

Author: rootonchair

src/diffusers/models/autoencoders/autoencoder_kl.py

@@ -300,7 +300,7 @@ def decode(self, z: torch.Tensor, return_dict: bool = True, generator=None) -> U
             decoded_slices = [self._decode(z_slice).sample for z_slice in z.split(1)]
             decoded = torch.cat(decoded_slices)
         else:
-            decoded = self._decode(z).sample
+            decoded = self._decode(z, return_dict=False)[0]
 
         if not return_dict:
             return (decoded,)