Skip to content

Commit 33c52de

Browse files
exclude transform_config from quantization_config parse (#421)
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent c1f0c25 commit 33c52de

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def from_pretrained_model(
207207

208208
@staticmethod
209209
def parse_sparsity_config(
210-
compression_config: Union[Dict[str, Any], "CompressedTensorsConfig"]
210+
compression_config: Union[Dict[str, Any], "CompressedTensorsConfig"],
211211
) -> Union[Dict[str, Any], None]:
212212
"""
213213
Parse sparsity config from quantization/compression config. Sparsity
@@ -227,7 +227,7 @@ def parse_sparsity_config(
227227

228228
@staticmethod
229229
def parse_quantization_config(
230-
compression_config: Union[Dict[str, Any], "CompressedTensorsConfig"]
230+
compression_config: Union[Dict[str, Any], "CompressedTensorsConfig"],
231231
) -> Union[Dict[str, Any], None]:
232232
"""
233233
Parse quantization config from quantization/compression config. The
@@ -246,6 +246,7 @@ def parse_quantization_config(
246246

247247
quantization_config = deepcopy(compression_config)
248248
quantization_config.pop(SPARSITY_CONFIG_NAME, None)
249+
quantization_config.pop(TRANSFORM_CONFIG_NAME, None)
249250

250251
# some fields are required, even if a qconfig is not present
251252
# pop them off and if nothing remains, then there is no qconfig

0 commit comments

Comments
 (0)