transformers
4005e30c - Fix incorrect attention mask truncate in WhisperFlashAttention2 (#36477)

Commit
1 year ago
Fix incorrect attention mask truncate in WhisperFlashAttention2 (#36477) * Fix incorrect attention mask truncate in whisper flash attention * also fix incorrect attention mask truncate in qwen2 audio * Nit attention mask truncate modeling_qwen2_audio.py * Nit attention mask truncate modeling_whisper.py Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> --------- Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com> Co-authored-by: eustlb <94853470+eustlb@users.noreply.github.com>
Author
Parents
Loading