File tree Expand file tree Collapse file tree 1 file changed +9
-7
lines changed
src/compressed_tensors/compressors/model_compressors Expand file tree Collapse file tree 1 file changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -201,9 +201,11 @@ def from_pretrained_model(
201
201
sparsity_config = sparsity_config ,
202
202
quantization_config = quantization_config ,
203
203
transform_config = transform_config ,
204
- compression_formats = [quantization_format ]
205
- if isinstance (quantization_format , str )
206
- else quantization_format ,
204
+ compression_formats = (
205
+ [quantization_format ]
206
+ if isinstance (quantization_format , str )
207
+ else quantization_format
208
+ ),
207
209
)
208
210
209
211
@staticmethod
@@ -314,10 +316,10 @@ def __init__(
314
316
315
317
self .quantization_compressor = {}
316
318
for format in self .compression_formats :
317
- self .quantization_compressor [
318
- format
319
- ] = BaseCompressor . load_from_registry (
320
- format , config = quantization_config
319
+ self .quantization_compressor [format ] = (
320
+ BaseCompressor . load_from_registry (
321
+ format , config = quantization_config
322
+ )
321
323
)
322
324
323
325
# ----- used by hf quantizer ----- #
You can’t perform that action at this time.
0 commit comments