fix tests

Author: a-r-r-o-w

src/diffusers/models/attention_processor.py

@@ -786,8 +786,8 @@ def __init__(
         in_channels: int,
         out_channels: int,
         num_attention_heads: Optional[int] = None,
-        mult: float = 1.0,
         attention_head_dim: int = 8,
+        mult: float = 1.0,
         norm_type: str = "batch_norm",
         kernel_sizes: Tuple[int, ...] = (5,),
         eps: float = 1e-15,

src/diffusers/models/autoencoders/autoencoder_dc.py

@@ -631,8 +631,8 @@ def tiled_decode(self, z: torch.Tensor, return_dict: bool = True) -> Union[Decod
         raise NotImplementedError("`tiled_decode` has not been implemented for AutoencoderDC.")
 
     def forward(self, sample: torch.Tensor, return_dict: bool = True) -> torch.Tensor:
-        z = self.encode(sample)
-        dec = self.decode(z)
+        z = self.encode(sample, return_dict=False)[0]
+        decoded = self.decode(z, return_dict=False)[0]
         if not return_dict:
-            return (dec,)
-        return DecoderOutput(sample=dec)
+            return (decoded,)
+        return DecoderOutput(sample=decoded)