stablediffusion
stablediffusion copied to clipboard
Wrong dim setting in SpatialTransformer
https://github.com/Stability-AI/stablediffusion/blob/d55bcd4d31d0316fcbdf552f2fd2628fdc812500/ldm/modules/attention.py#L318
Since the proj_in does in_ch->inner_dim Shouldn't the proj_out does inner_dim -> in_ch?
The conv version of proj_out also does inner_dim -> in_ch
https://github.com/Stability-AI/stablediffusion/blob/d55bcd4d31d0316fcbdf552f2fd2628fdc812500/ldm/modules/attention.py#L312-L316
I also noticed this recently. This is an error I think.