def forward(self, x: torch.Tensor):
        """Forward pass through the residual attention block."""
        x = x + self.attention(self.ln_1(x))
        x = x + self.mlp(self.ln_2(x))
        return x
Forward pass through the residual attention block.
examples/community/pipeline_faithdiff_stable_diffusion_xl.py