transformers
Wrap `_prepare_4d_causal_attention_mask` as a leaf function
#27236

Merged

Wrap `_prepare_4d_causal_attention_mask` as a leaf function #27236

amyeroberts merged 1 commit into huggingface:main from michaelbenayoun:fx_for_for_pp_with_llama

Wrap _prepare_4d_causal_attention_mask as a leaf function

b47bf503

michaelbenayoun requested a review from

younesbelkada 2 years ago

amyeroberts approved these changes on 2023-11-02

younesbelkada commented on 2023-11-02

younesbelkada approved these changes on 2023-11-02

amyeroberts merged 4557a0de into main 2 years ago

michaelbenayoun deleted the fx_for_for_pp_with_llama branch 2 years ago

Reviewers

younesbelkada

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone