Bug in the Spatial Attention class.

[faruknane]

Hi,

here in this line, there is a minor bug.

Instead of self.proj_out = zero_module(nn.Linear(in_channels, inner_dim)) it should be self.proj_out = zero_module(nn.Linear(inner_dim, in_channels))