transformers
Add zero dim tensor check when using flash_attention
#38280

Merged

Add zero dim tensor check when using flash_attention #38280

ArthurZucker merged 2 commits into huggingface:main from ranzhejiang:zhejiang_fix_flash_attn

ranzhejiang force pushed from 007835f0 to 6afac400 324 days ago

ranzhejiang force pushed from 6afac400 to a9fdcf48 324 days ago

ArthurZucker commented on 2025-05-23

ranzhejiang force pushed from a9fdcf48 to 007835f0 316 days ago

ranzhejiang force pushed from 007835f0 to 3d367724 316 days ago

ranzhejiang force pushed from 3d367724 to 8365960c 316 days ago

Add zero dim tensor check when using flash_attention

8365960c

Add zero dim tensor check when using flash_attention

827b2304

ArthurZucker merged ae32f1ad into main 290 days ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone