Skip to content
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

Allow torch 2.3 and remove torch ceiling version restriction #2259

Merged
merged 7 commits into from
May 13, 2024

Conversation

mgoin
Copy link
Member

@mgoin mgoin commented Apr 30, 2024

vLLM now requires torch 2.3.0, so we should try to raise the restriction in SparseML. Going forward, I think we shouldn't be so restrictive to newer versions of pytorch and should raise a warning at most, not an explicit exception.

rahul-tuli
rahul-tuli previously approved these changes Apr 30, 2024
dbogunowicz
dbogunowicz previously approved these changes May 2, 2024
@mgoin
Copy link
Member Author

mgoin commented May 2, 2024

@bfineran @robertgshaw2-neuralmagic is this good with you going forward to not restrict pytorch?

bfineran
bfineran previously approved these changes May 6, 2024
@mgoin mgoin mentioned this pull request May 7, 2024
8 tasks
@dbogunowicz
Copy link
Contributor

@mgoin should we land this (after the GHA tests have passed?)

@mgoin mgoin dismissed stale reviews from dbogunowicz, rahul-tuli, and bfineran via 1d6091e May 13, 2024 17:39
@mgoin mgoin merged commit c234638 into main May 13, 2024
17 checks passed
@mgoin mgoin deleted the upgrade-torch-2.3 branch May 13, 2024 18:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

5 participants