src/diffusers/models/attention.py · 5adb0a7bf7fee1732f307dfda9013bf2cf7b0ff9 · renzhc / diffusers_dcu · GitLab

Find file Blame History Permalink

use torch.matmul instead of einsum in attnetion. (#445) · 5adb0a7b
Suraj Patil authored Sep 09, 2022
```
* use torch.matmul instead of einsum

* fix softmax
```
5adb0a7b

attention.py 13 KB