Skip to content

Commit 1c41310

Browse files
authored
[Bugfix] Fix transform_config parsing in Compressed Tensors (#23945)
Signed-off-by: Kyle Sayers <[email protected]>
1 parent c83c4ff commit 1c41310

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def __init__(
6363
sparsity_ignore_list: list[str],
6464
kv_cache_scheme: Optional[dict[str, Any]] = None,
6565
config: Optional[dict[str, Any]] = None,
66-
transform_config: Optional[TransformConfig] = None,
66+
transform_config: Optional[dict[str, Any]] = None,
6767
):
6868
super().__init__()
6969
self.ignore = ignore
@@ -75,7 +75,7 @@ def __init__(
7575
self.sparsity_ignore_list = sparsity_ignore_list
7676
self.config = config
7777

78-
if transform_config is not None:
78+
if transform_config:
7979
self.transform_config = TransformConfig.model_validate(
8080
transform_config)
8181
else:

0 commit comments

Comments
 (0)