diff --git a/src/sparsetensors/utils/__init__.py b/src/sparsetensors/utils/__init__.py index 5bc0fec2..e9e78d44 100644 --- a/src/sparsetensors/utils/__init__.py +++ b/src/sparsetensors/utils/__init__.py @@ -13,4 +13,5 @@ # limitations under the License. # flake8: noqa +from .helpers import * from .safetensors_load import * diff --git a/src/sparsetensors/utils/helpers.py b/src/sparsetensors/utils/helpers.py index 8d06cde2..c584c2ee 100644 --- a/src/sparsetensors/utils/helpers.py +++ b/src/sparsetensors/utils/helpers.py @@ -15,9 +15,9 @@ from typing import Optional +from sparsetensors.base import SPARSITY_CONFIG_NAME from sparsetensors.compressors import ModelCompressor from sparsetensors.config import CompressionConfig -from sparsetensors.utils.helpers import SPARSITY_CONFIG_NAME from transformers import AutoConfig