Skip to content

Commit

Permalink
Update transformer-engine[pytorch] requirement
Browse files Browse the repository at this point in the history
Updates the requirements on [transformer-engine[pytorch]](https://github.com/NVIDIA/TransformerEngine) to permit the latest version.
- [Release notes](https://github.com/NVIDIA/TransformerEngine/releases)
- [Commits](NVIDIA/TransformerEngine@v1.11...v1.12)

---
updated-dependencies:
- dependency-name: transformer-engine[pytorch]
  dependency-type: direct:development
...

Signed-off-by: dependabot[bot] <[email protected]>
  • Loading branch information
dependabot[bot] authored Dec 2, 2024
1 parent 41e02d7 commit 0e8db24
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@
]

extra_deps['te'] = [
'transformer-engine[pytorch]>=1.11.0,<1.12',
'transformer-engine[pytorch]>=1.11.0,<1.13',
]

extra_deps['databricks-serverless'] = {
Expand Down

0 comments on commit 0e8db24

Please sign in to comment.