Fix: Unexpected Keys, Improve run_compressed, Rename Test Folder (#37077)

This commit is contained in:
Rahul Tuli
2025-04-04 14:30:11 -05:00
committed by GitHub
parent 531e4fcf0e
commit ebe47ce3e9
6 changed files with 48 additions and 32 deletions

View File

@@ -185,6 +185,7 @@ class RunCompressedTest(unittest.TestCase):
def test_default_run_compressed__False(self):
from compressed_tensors.linear.compressed_linear import CompressedLinear
from compressed_tensors.quantization.utils import iter_named_leaf_modules
from transformers.utils.quantization_config import CompressedTensorsConfig
quantization_config = CompressedTensorsConfig(run_compressed=False)