def forward(self, x_noise, t, x_start, y, batch_mul=1):
        """
        Forward pass of ransformer.
        x: (N, C, H, W) tensor of spatial inputs (images or latent representations of images)
        t: (N,) tensor of diffusion timesteps
        y: (N,) tensor of class labels
        """
        condition = self.forward_parallel(x_start, y)
        condition = condition.repeat_interleave(batch_mul, dim=0)
        x = self.forward_diffusion(x_noise, t, condition)
        return x

        Forward pass of ransformer.
        x: (N, C, H, W) tensor of spatial inputs (images or latent representations of images)
        t: (N,) tensor of diffusion timesteps
        y: (N,) tensor of class labels
        
LatentLM/models/Transformer.py