DeepSpeed
00757a19 - Fix rope_theta arg for diffusers_attention (#4656)

Commit
2 years ago
Fix rope_theta arg for diffusers_attention (#4656) This PR updates `diffusers_attention` to properly pass the `rope_theta` arg to the `linear_func` calls. This was added in GH-4480 and needed to be updated for the diffusers attention module as well. Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
Author
Parents
Loading