transformers
4557a0de - Wrap `_prepare_4d_causal_attention_mask` as a leaf function (#27236)

Commit

2 years ago

Wrap `_prepare_4d_causal_attention_mask` as a leaf function (#27236) Wrap _prepare_4d_causal_attention_mask as a leaf function

References

Author

michaelbenayoun

Parents