initial commit

neuralmagic · Apr 8, 2024 · 097bd79 · 097bd79
1 parent 5aae81b
commit 097bd79
Show file tree

Hide file tree

Showing 18 changed files with 40 additions and 782 deletions.
diff --git a/src/sparseml/transformers/compression/README.md b/src/sparseml/transformers/compression/README.md
@@ -1,3 +1,4 @@
+// # TODO: Edit this readme
 # Save/Load Compressed SafeTensors
 
 ## Motivation

diff --git a/src/sparseml/transformers/compression/__init__.py b/src/sparseml/transformers/compression/__init__.py
@@ -13,6 +13,4 @@
 # limitations under the License.
 
 # flake8: noqa
-
-from .compressors import *
 from .config import *
diff --git a/src/sparseml/transformers/compression/compressors/__init__.py b/src/sparseml/transformers/compression/compressors/__init__.py
diff --git a/src/sparseml/transformers/compression/compressors/base.py b/src/sparseml/transformers/compression/compressors/base.py
diff --git a/src/sparseml/transformers/compression/compressors/dense.py b/src/sparseml/transformers/compression/compressors/dense.py
diff --git a/src/sparseml/transformers/compression/compressors/sparse_bitmask.py b/src/sparseml/transformers/compression/compressors/sparse_bitmask.py
diff --git a/src/sparseml/transformers/compression/config/__init__.py b/src/sparseml/transformers/compression/config/__init__.py
@@ -14,6 +14,6 @@
 
 # flake8: noqa
 
-from .base import CompressionConfig
-from .dense import DenseSparsityConfig
-from .sparse_bitmask import BitmaskConfig
+from .base import SparseMLCompressionConfig
+from .dense import SparseMLDenseSparsityConfig
+from .sparse_bitmask import SparseMLBitmaskConfig