Wrap _prepare_4d_causal_attention_mask as a leaf function#27236
Merged
amyeroberts merged 1 commit intohuggingface:mainfrom Nov 2, 2023
Merged
Wrap _prepare_4d_causal_attention_mask as a leaf function#27236amyeroberts merged 1 commit intohuggingface:mainfrom
_prepare_4d_causal_attention_mask as a leaf function#27236amyeroberts merged 1 commit intohuggingface:mainfrom
