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 Oct 15, 2024
1 parent ce514db commit 335f29e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions neural_compressor/torch/algorithms/weight_only/save_load.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,6 +230,7 @@ def load_hf_format_woq_model(self):
): # # pragma: no cover
# load autoround format quantized model
from auto_round import AutoRoundConfig

hf_kargs = {}
pretrain_args = ["trust_remote_code", "_attn_implementation", "device_map", "torch_dtype"]
for item in pretrain_args:
Expand Down Expand Up @@ -463,7 +464,7 @@ def _get_model_class_and_config(self):
if model_cls:
model_class = model_cls
else:
logger.info(f"Could't find model class.")
logger.info("Couldn't find model class.")
return model_class, config

def _get_loaded_state_dict(self, config):
Expand Down Expand Up @@ -921,4 +922,3 @@ def _use_hpu_module(self): # pragma: no cover
if os.path.exists(os.path.join(self._model_local_dir, HPU_WEIGHT_NAME)):
return True
return False

0 comments on commit 335f29e

Please sign in to comment.