src/diffusers/models/embeddings.py · b9e2f886cd6e9182f1bf1bf7421c6363956f94c5 · renzhc / diffusers_dcu

asfiyab-nvidia authored Sep 10, 2024



Remove Squeeze op
Signed-off-by: Asfiya Baig <asfiyab@nvidia.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>

b9e2f886

embeddings.py 66.5 KB