stablediffusion icon indicating copy to clipboard operation
stablediffusion copied to clipboard

Wrong dim setting in SpatialTransformer

Open KohakuBlueleaf opened this issue 2 years ago • 1 comments

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

KohakuBlueleaf avatar Dec 16 '22 03:12 KohakuBlueleaf

I also noticed this recently. This is an error I think.

daveboat avatar Jun 22 '23 13:06 daveboat