diff --git a/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py b/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py index b07bf675ca47..97041a5a050f 100644 --- a/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py +++ b/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py @@ -63,7 +63,7 @@ def __init__( sparsity_ignore_list: list[str], kv_cache_scheme: Optional[dict[str, Any]] = None, config: Optional[dict[str, Any]] = None, - transform_config: Optional[TransformConfig] = None, + transform_config: Optional[dict[str, Any]] = None, ): super().__init__() self.ignore = ignore @@ -75,7 +75,7 @@ def __init__( self.sparsity_ignore_list = sparsity_ignore_list self.config = config - if transform_config is not None: + if transform_config: self.transform_config = TransformConfig.model_validate( transform_config) else: