Fix import of transformmodindex on nightly - Githubissues

pytorch-labs / attention-gym

Helpful tools and examples for working with flex-attention

BSD 3-Clause "New" or "Revised" License

484 stars 24 forks source link

Fix import of transformmodindex on nightly #52

Closed drisspg closed 1 month ago

drisspg commented 1 month ago

Stacked PRs:

->#52

Fix Import issue

Location moved in https://github.com/pytorch/pytorch/commit/b3f30c9bc330ba3bbaa951794dbda8df2b43e326#diff-82826ad59d66d1141586434e8a[…]a0ee9e94020493af5f76f1d161f3e55