sdpa: support attn_mask.requires_grad, support expanded number of heads in attn_mask #2036
Annotations
1 warning
|
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
|
Loading