diff --git a/src/compressed_tensors/utils/helpers.py b/src/compressed_tensors/utils/helpers.py index db77bccb..9ec522e0 100644 --- a/src/compressed_tensors/utils/helpers.py +++ b/src/compressed_tensors/utils/helpers.py @@ -12,12 +12,16 @@ # See the License for the specific language governing permissions and # limitations under the License. -from typing import Any, Dict, Optional +from typing import TYPE_CHECKING, Any, Dict, Optional import torch from transformers import AutoConfig +if TYPE_CHECKING: + from compressed_tensors.compressors import ModelCompressor + + __all__ = [ "infer_compressor_from_model_config", "fix_fsdp_module_name",