Skip to content

Commit 829b7cb

Browse files
formatting
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent 606f177 commit 829b7cb

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -201,9 +201,11 @@ def from_pretrained_model(
201201
sparsity_config=sparsity_config,
202202
quantization_config=quantization_config,
203203
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+
),
207209
)
208210

209211
@staticmethod
@@ -314,10 +316,10 @@ def __init__(
314316

315317
self.quantization_compressor = {}
316318
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+
)
321323
)
322324

323325
# ----- used by hf quantizer ----- #

0 commit comments

Comments
 (0)