fix arg passed error in CPAttention

Tangshitao / MVDiffusion

MVDiffusion: Enabling Holistic Multi-view Image Generation with Correspondence-Aware Diffusion, NeurIPS 2023 (spotlight)

447 stars 21 forks source link

fix arg passed error in CPAttention #24

Closed iaoqian closed 9 months ago

iaoqian commented 9 months ago

Hi Tang, thanks for sharing these detailed code. But I I'm pretty sure this is a tiny yet true mistake in this part of code which makes an error when I customize my for data with different shape for this favourable work.

Tangshitao commented 9 months ago

Thanks for the correction.