-
Support TE-DPA For Stable Diffusion (#10314)
frostbyte_neo 发布于
2024-09-16 09:16:09 +00:00 - [SD] Add te-dpa support
Signed-off-by: Wil Kong alpha0422@gmail.com
- [SD] Add te-dpa support, resolve compatibility with TE-master
Signed-off-by: Wil Kong alpha0422@gmail.com
- [SD] Add te-dpa support, add check for attention configs.
Signed-off-by: Wil Kong alpha0422@gmail.com
- Fix bugs of flash-attn and dpa in SD.
Signed-off-by: Wil Kong alpha0422@gmail.com
- Fix the issue of DPA API change.
Signed-off-by: Wil Kong alpha0422@gmail.com
- Apply isort and black reformatting
Signed-off-by: alpha0422 alpha0422@users.noreply.github.com
Signed-off-by: Wil Kong alpha0422@gmail.com- [SD] TE-DPA: disbale use te-dpa in inference flow.
Signed-off-by: Wil Kong alpha0422@gmail.com
Signed-off-by: alpha0422 alpha0422@users.noreply.github.com
Co-authored-by: Mengdi Wang didow@nvidia.com下载附件