Class LinAttnBlock

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

to match AttnBlock usage

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):