-
Notifications
You must be signed in to change notification settings - Fork 26.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add sdpa for Vivit #33757
Add sdpa for Vivit #33757
Conversation
@amyeroberts All green 🟢 😃 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @amyeroberts , I think in the One thing can be that apart from the Or the other thing can be that I fix it here in this PR. Please suggest what works well for you?! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
looks good to me, appreciate the effort @RUFFY-369
Update: This and this commit was enough to fix the interpolate method and the only thing required right now is a nit in And regarding this PR,
Both of the things asked has been completed and pushed. So, this PR is ready for merge. |
Hi @amyeroberts , I just created a separate PR where Please review both of them when you get the time. |
@amyeroberts Can you please add |
@RUFFY-369 You should be able to add the label yourself, I think. Could you try? It would be useful to know if external collaborators can or can't add this, as we'd like to make this so that contributors can do as much as possible independently |
@amyeroberts No, I can't. On my side I don't have any gear/settings button near label: Usually if I can add the label, I would see something like this: So, yes, I guess external collaborators can't add this |
@RUFFY-369 OK, thanks for checking! @ydshieh This means the message for the slow model tests bot might need to be changed as contributors can't add the label as instructed |
@RUFFY-369 I've added the label. Could you push another |
Hey! 🤗 Thanks for your contribution to the Before merging this pull request, slow tests CI should be triggered. To enable this:
(For maintainers) The documentation for slow tests CI on PRs is here. |
@amyeroberts Done 👍 |
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
Failing tests are unrelated |
@RUFFY-369 Could you rebase? A fix has been merged in |
@amyeroberts Done |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for adding @RUFFY-369!
The only thing missing is a fallback for when output_attentions are passed e.g. like here
cc @qubvel
@amyeroberts Fallback has been added for Cheers! |
All green, slow tests are left to trigger |
soft ping @amyeroberts |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, looks good to me! cc @ArthurZucker for final review
soft ping @ArthurZucker |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM thanks for the details documentation update!
🤗 |
* chore:add sdpa to vivit * fix:failing slow test_inference_interpolate_pos_encoding(failing on main branch too) * chore:fix nits * ci:fix repo consistency failure * chore:add info and benchmark to model doc * [run_slow] vivit * chore:revert interpolation test fix for new issue * [run_slow] vivit * [run_slow] vivit * [run_slow] vivit * chore:add fallback for output_attentions being True * [run_slow] vivit * style:make fixup * [run_slow] vivit
What does this PR do?
Towards #28005
Adds sdpa for the ViViT model.
Before submitting
Pull Request section?
to it if that's the case.
documentation guidelines, and
here are tips on formatting docstrings.
Who can review?
@amyeroberts