Code

[WIP] Add flex attention for gpt2#34861

Code

Draft

mayankagarwals wants to merge 2 commits intohuggingface:mainhuggingface/transformers:mainfrom mayankagarwals:gpt2_flexmayankagarwals/transformers:gpt2_flexCopy head branch name to clipboard

+2-224

Commits

Commits on Nov 21, 2024

• initial commit

  mayankagarwals

  committed
  08ee8bc

  Copy full SHA for 08ee8bc

Commits on Dec 6, 2024

• Merge branch 'main' into gpt2_flex

  mayankagarwals

  authored
  29376a4

  Copy full SHA for 29376a4