flax
2fae1bad - Add `attention_bias` parameter to `MultiHeadDotProductAttention`.

Commit

1 year ago

Add `attention_bias` parameter to `MultiHeadDotProductAttention`. PiperOrigin-RevId: 746547988

References

Author

Jan Hosang

Committer

a-googler

Parents