Skip to content

Commit b9101e6

Browse files
committed
fix docstring
Signed-off-by: Xin He <xinhe3@habana.ai>
1 parent 5d7667a commit b9101e6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

neural_compressor/torch/algorithms/weight_only/save_load.py

+1
Original file line numberDiff line numberDiff line change
@@ -956,6 +956,7 @@ def _use_hpu_module(self): # pragma: no cover
956956

957957

958958
def change_config_to_hf_format(config_mappings):
959+
"""Change INC config_mappings to Huggingface format."""
959960
# Refer to https://huggingface.co/TheBloke/Llama-2-7B-Chat-GPTQ/blob/main/config.json
960961
default_quantization_config = {
961962
"bits": 4,

0 commit comments

Comments
 (0)