Skip to content

Commit 90cf72d

Browse files
committed
style
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
1 parent 48fbbd8 commit 90cf72d

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -517,9 +517,7 @@ def _replace_weights(self, dense_weight_generator, model: Module):
517517
update_parameter_data(module, data, param_name)
518518

519519

520-
def map_module_to_scheme(
521-
model: Module,
522-
) -> Dict[str, QuantizationScheme]:
520+
def map_module_to_scheme(model: Module) -> Dict[str, QuantizationScheme]:
523521
"""
524522
Given a pytorch model, map out the submodule name (usually linear layers)
525523
to the weight QuantizationArgs. If running input activation quantization, will also

0 commit comments

Comments
 (0)