search

neuralmagic / sparseml

Libraries for applying sparsification recipes to neural networks with a few lines of code, enabling faster and smaller models
Apache License 2.0
2.07k stars 148 forks source link

SparseML dependency on `compressed-tensors` #2229

Closed dbogunowicz closed 7 months ago

dbogunowicz commented 7 months ago

GHA will be green after: https://github.com/neuralmagic/compressed-tensors/pulls lands

Satrat commented 7 months ago

@bfineran @dbogunowicz is there anything blocking this from being merged??

dbogunowicz commented 7 months ago

@Satrat I'd say the readiness of the pypi package. This PR is currently compatible with the main branch of compressed-tensors.

mgoin commented 7 months ago

@dbogunowicz @Satrat I would also mention pydantic v2 support as a blocker https://github.com/neuralmagic/sparseml/pull/2248

rahul-tuli commented 7 months ago