fix style

Signed-off-by: Kyle Sayers <[email protected]>

Author: kylesayrs

src/compressed_tensors/quantization/quant_config.py

@@ -171,7 +171,7 @@ def from_pretrained(
         :param model: model to calculate quantization scheme of
         :return: filled out QuantizationScheme for the input model
         """
-        from compressed_tensors.modeling import IMPL_ATTR, KV_CACHE_ATTR
+        from compressed_tensors.modeling import IMPL_ATTR
         from compressed_tensors.quantization.lifecycle.initialize import (
             is_attention_module,
         )

tests/test_quantization/test_quant_config.py

@@ -16,14 +16,11 @@
 from compressed_tensors.quantization import (
     DEFAULT_QUANTIZATION_FORMAT,
     DEFAULT_QUANTIZATION_METHOD,
-    QuantizationArgs,
     QuantizationConfig,
     QuantizationScheme,
     QuantizationStatus,
-    apply_quantization_config,
 )
 from pydantic import ValidationError
-from transformers import AutoModelForCausalLM
 
 
 def test_basic_config():