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](https://github.com/NVIDIA/TransformerEngine/commits)

---
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 9, 2024
1 parent 7b8bf5f commit d059397
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.14',
]

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

0 comments on commit d059397

Please sign in to comment.