stablediffusion
stablediffusion copied to clipboard
Fix wrong dim setting in SpatialTransformer
Reference to this issue: https://github.com/Stability-AI/stablediffusion/issues/104
The proj_out should convert inner_dim back to in_channels.