Code

[ESM] Add support for sdpa.#34954

Code

Open

wzf03 wants to merge 4 commits intohuggingface:mainhuggingface/transformers:mainfrom wzf03:esm-sdpa-supportwzf03/transformers:esm-sdpa-supportCopy head branch name to clipboard

+398-34

Commits

Commits on Nov 28, 2024

• Add SDPA support for ESM.

  wzf03

  committed
  72750f6

• Add SDPA test for ESMFold.

  wzf03

  committed
  996880a

Commits on Nov 29, 2024

• [run-slow] esm

  Rocketknight1

  committed
  0ddfe44

Commits on Dec 2, 2024

• Explicitly move input_ids to desired device in test_modeling_esm.

  wzf03

  committed
  f5d9ecc