sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #5327
Annotations
1 warning
|
The following artifacts were uploaded using a version of actions/upload-artifact that is scheduled for deprecation: "dist-packages-37e65ffeb7bc0cadee895f2dba68ad5e0c40838f".
Please update your workflow to use v4 of the artifact actions.
Learn more: https://github.blog/changelog/2024-04-16-deprecation-notice-v3-of-the-artifact-actions/
|
Loading