SparseML dependency on compressed-tensors (#2229)

* initial commit

* update setup.py

* Update setup.py

* fix setup.py

* move all config to sparsetensors

* cleanup class name and comments

* refactor to compressed-tensors

* all tests passing

* remove unused config

* bring back SparsityConfigMetadata

* Update setup.py

Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>

* update setup

* fix import problem

* fix clearml test

* compressed-tensors are transformers dep

* update compatibility to 0.3.2

---------

Co-authored-by: Sara Adkins <sara@neuralmagic.com>
Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>

Author: 3 people

setup.py

@@ -87,6 +87,7 @@
     "evaluate>=0.4.1",
     "accelerate>=0.20.3",
     "safetensors>=0.4.1",
+    "compressed-tensors",
 ]
 _llm_deps = _transformers_deps + ["sentencepiece"]
 _yolov5_deps = _pytorch_vision_deps + [

src/sparseml/transformers/compression/README.md

@@ -11,8 +11,3 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-# flake8: noqa
-
-from .compressors import *
-from .config import *