(self, in_channels)
| 23 | |
| 24 | class AttnBlock: |
| 25 | def __init__(self, in_channels): |
| 26 | self.norm = GroupNorm(32, in_channels) |
| 27 | self.q = Conv2d(in_channels, in_channels, 1) |
| 28 | self.k = Conv2d(in_channels, in_channels, 1) |
| 29 | self.v = Conv2d(in_channels, in_channels, 1) |
| 30 | self.proj_out = Conv2d(in_channels, in_channels, 1) |
| 31 | |
| 32 | # copied from AttnBlock in ldm repo |
| 33 | def __call__(self, x): |