@@ -207,7 +207,7 @@ def from_pretrained_model(
207
207
208
208
@staticmethod
209
209
def parse_sparsity_config (
210
- compression_config : Union [Dict [str , Any ], "CompressedTensorsConfig" ]
210
+ compression_config : Union [Dict [str , Any ], "CompressedTensorsConfig" ],
211
211
) -> Union [Dict [str , Any ], None ]:
212
212
"""
213
213
Parse sparsity config from quantization/compression config. Sparsity
@@ -227,7 +227,7 @@ def parse_sparsity_config(
227
227
228
228
@staticmethod
229
229
def parse_quantization_config (
230
- compression_config : Union [Dict [str , Any ], "CompressedTensorsConfig" ]
230
+ compression_config : Union [Dict [str , Any ], "CompressedTensorsConfig" ],
231
231
) -> Union [Dict [str , Any ], None ]:
232
232
"""
233
233
Parse quantization config from quantization/compression config. The
@@ -246,6 +246,7 @@ def parse_quantization_config(
246
246
247
247
quantization_config = deepcopy (compression_config )
248
248
quantization_config .pop (SPARSITY_CONFIG_NAME , None )
249
+ quantization_config .pop (TRANSFORM_CONFIG_NAME , None )
249
250
250
251
# some fields are required, even if a qconfig is not present
251
252
# pop them off and if nothing remains, then there is no qconfig
0 commit comments