Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Feb 11, 2025
1 parent 558b734 commit c0b0632
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/ds/quant.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def quant_model_weight_with_low_cpu_usage(model_path, qmodel_path):
for weight_name in f.keys():
weight = f.get_tensor(weight_name)
if skip_weight(weight_name):
logger.debug(f"Skiping quantize {weight_name}")
logger.debug(f"Skipping quantize {weight_name}")
qtensors[weight_name] = weight
qtensor_mappping[weight_name] = qmodel_file_name
continue
Expand Down Expand Up @@ -141,7 +141,7 @@ def static_quant_model_tran(model_path, qmodel_path):
)
for name, module in model.named_modules():
if not isinstance(module, QUANT_MODULE_TYPES) or skip_weight(name):
logger.debug(f"Skiping quantize {name}")
logger.debug(f"Skipping quantize {name}")
continue
logger.debug(f"Processing {name}")
weight = module.weight
Expand Down

0 comments on commit c0b0632

Please sign in to comment.