sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard

Allow torch 2.3 and remove torch ceiling version restriction

Open mgoin opened this issue 1 year ago • 2 comments

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.

mgoin avatar Apr 30 '24 17:04 mgoin

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

mgoin avatar May 02 '24 13:05 mgoin

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

dbogunowicz avatar May 13 '24 08:05 dbogunowicz