transformers
Wrap `_prepare_4d_causal_attention_mask` as a leaf function
#27236
Merged
Merged
Wrap `_prepare_4d_causal_attention_mask` as a leaf function #27236
asartran
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
asartran
approved these changes
on 2023-11-02
younesbelkada
commented
on 2023-11-02
younesbelkada
approved these changes
on 2023-11-02
Login to write a write a comment.
Login via GitHub