-
Notifications
You must be signed in to change notification settings - Fork 31.9k
Add torch >=1.12 requirement for Tapas
#24251
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
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
amyeroberts
left a comment
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 fixing!
Co-authored-by: amyeroberts <[email protected]>
| @unittest.skipIf( | ||
| not is_torch_greater_or_equal_than_1_12, | ||
| reason="`GPTBigCode` checkpoints use `PytorchGELUTanh` which requires `torch>=1.12.0`." | ||
| ) |
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.
@amyeroberts Just put this one to this PR to not opening a new PR. Sorry for doing this after PR being reviewed.
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.
No worries, thanks for pinging me on the comment to highlight :)
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.
Approval still stands :)
…/transformers into fix_tapas_in_past_ci
* fix * fix * fix * Update src/transformers/models/tapas/modeling_tapas.py Co-authored-by: amyeroberts <[email protected]> * fix --------- Co-authored-by: ydshieh <[email protected]> Co-authored-by: amyeroberts <[email protected]>
What does this PR do?
Tapas files are changed in #20149 to use torch's
scatter. The torch tensor's methodscatter_reduceaccept the argumentsrconly for torch >= 1.12.This PR add some warnings/requirements in tapas modeling/test files to avoid test failures in past CI with torch <= 1.11.
(one previous similar PR is #19851)