Method __init__

code/dc_ldm/modules/diffusionmodules/model.py:146–147

(self, in_channels)

class LinAttnBlock(LinearAttention):
    """to match AttnBlock usage"""
    def __init__(self, in_channels):
        super().__init__(dim=in_channels, heads=1, dim_head=in_channels)


class AttnBlock(nn.Module):