-
Notifications
You must be signed in to change notification settings - Fork 27.3k
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 type hints for several pytorch models (batch-2) #25557
Add type hints for several pytorch models (batch-2) #25557
Conversation
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.
Overall, this looks good! I raised one issue with the attention_mask
types, but that's it. Note that there are probably a few other instances with the wrong attention_mask
type, though I only indicated one in the review.
src/transformers/models/decision_transformer/modeling_decision_transformer.py
Outdated
Show resolved
Hide resolved
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
This looks good to me now! Ping me whenever you're finished and you want me to merge it. |
Matt, this is fine by me so I'm happy for you to merge it |
* Add missing type hint to cpmant * Add type hints to decision_transformer model * Add type hints to deformable_detr models * Add type hints to detr models * Add type hints to deta models * Add type hints to dpr models * Update attention mask type hint Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> * Update remaining attention masks type hints * Update docstrings' type hints related to attention masks --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
* Add missing type hint to cpmant * Add type hints to decision_transformer model * Add type hints to deformable_detr models * Add type hints to detr models * Add type hints to deta models * Add type hints to dpr models * Update attention mask type hint Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> * Update remaining attention masks type hints * Update docstrings' type hints related to attention masks --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
* Add missing type hint to cpmant * Add type hints to decision_transformer model * Add type hints to deformable_detr models * Add type hints to detr models * Add type hints to deta models * Add type hints to dpr models * Update attention mask type hint Co-authored-by: Matt <Rocketknight1@users.noreply.github.com> * Update remaining attention masks type hints * Update docstrings' type hints related to attention masks --------- Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
What does this PR do?
Addresses some of the models in #16059
Who can review?
@Rocketknight1, please