remove unused utils

Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>

Author: kylesayrs

src/compressed_tensors/compressors/model_compressors/model_compressor.py

@@ -33,15 +33,13 @@
 from compressed_tensors.compressors.base import BaseCompressor
 from compressed_tensors.compressors.sparse_compressors import DenseCompressor
 from compressed_tensors.config import CompressionFormat, SparsityCompressionConfig
-from compressed_tensors.linear.compressed_linear import CompressedLinear
 from compressed_tensors.quantization import (
     DEFAULT_QUANTIZATION_METHOD,
     QuantizationConfig,
     QuantizationScheme,
     QuantizationStatus,
     apply_quantization_config,
     load_pretrained_quantization_parameters,
-    unwrap_module_forward_quantized,
 )
 from compressed_tensors.quantization.lifecycle import expand_target_names
 from compressed_tensors.quantization.utils import (
@@ -52,15 +50,13 @@
     get_safetensors_folder,
     has_offloaded_params,
     merge_names,
-    module_map_replace,
     register_offload_parameter,
     update_parameter_data,
 )
 from compressed_tensors.utils.helpers import (
     fix_fsdp_module_name,
     is_compressed_tensors_config,
 )
-from compressed_tensors.utils.offload import disable_hf_hook, update_offload_parameter
 from torch import Tensor
 from torch.nn import Module
 from tqdm import tqdm

src/compressed_tensors/utils/helpers.py

@@ -14,11 +14,10 @@
 
 import warnings
 from functools import wraps
-from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional
 
 import numpy
 import torch
-import tqdm
 from transformers import AutoConfig
 
 
@@ -39,8 +38,6 @@
     "shard_tensor",
     "pack_bitmasks",
     "unpack_bitmasks",
-    "remove_suffix",
-    "module_map_replace",
 ]
 
 FSDP_WRAPPER_NAME = "_fsdp_wrapped_module"
@@ -331,48 +328,3 @@ def unpack_bitmasks(
     )
 
     return unpacked_bitmasks_torch
-
-
-def remove_suffix(value: str, suffix: str) -> str:
-    # can replace with str.removesuffix in python3.9+
-    assert value.endswith(suffix)
-    return value[: -len(suffix)]
-
-
-def module_map_replace(
-    module: torch.nn.Module,
-    func: Callable[[torch.nn.Module], torch.nn.Module],
-    progress: Union[bool, tqdm.tqdm] = False,
-    pre: bool = True,
-) -> torch.nn.Module:
-    """
-    Replaces modules in a given `torch.nn.Module` recursively using a provided function.
-
-    This function traverses the module hierarchy and applies the `func` transformation
-    either before (`pre=True`) or after (`pre=False`) recursing into children modules.
-    Optionally displays progress using tqdm.
-
-    :param module: root module to replace
-    :param func: module mapping function
-    :param progress: if True, display a tqdm progress bar.
-        If a `tqdm.tqdm` instance is provided, the instance will be updated
-    :param pre: if True, apply with pre-order, post-order otherwise
-    :return: the modified module after applying the function to all submodules
-    """
-    if progress is True:
-        total = len(list(module.modules()))
-        progress = tqdm.tqdm(total=total)
-
-    if pre:
-        module = func(module)
-
-    for name, child in list(module.named_children()):
-        module.add_module(name, module_map_replace(child, func, pre, progress))
-
-    if not pre:
-        module = func(module)
-
-    if isinstance(progress, tqdm.tqdm):
-        progress.update(1)
-
-    return module