Skip to content

Commit 944ebb1

Browse files
committed
fix vulnerability
Signed-off-by: xin3he <xin3.he@intel.com>
1 parent 3eb5529 commit 944ebb1

File tree

1 file changed

+2
-30
lines changed

1 file changed

+2
-30
lines changed

neural_compressor/utils/load_huggingface.py

+2-30
Original file line numberDiff line numberDiff line change
@@ -96,33 +96,8 @@ def from_pretrained(cls, model_name_or_path: str, **kwargs) -> torch.nn.Module:
9696
)
9797
return model
9898
else:
99-
logger.info("the quantization optimized model is loading.")
100-
keys_to_ignore_on_load_unexpected = copy.deepcopy(
101-
getattr(model_class, "_keys_to_ignore_on_load_unexpected", None)
102-
)
103-
keys_to_ignore_on_load_missing = copy.deepcopy(
104-
getattr(model_class, "_keys_to_ignore_on_load_missing", None)
105-
)
106-
107-
# Avoid unnecessary warnings resulting from quantized model initialization
108-
quantized_keys_to_ignore_on_load = [
109-
r"zero_point",
110-
r"scale",
111-
r"packed_params",
112-
r"constant",
113-
r"module",
114-
r"best_configure",
115-
]
116-
if keys_to_ignore_on_load_unexpected is None:
117-
model_class._keys_to_ignore_on_load_unexpected = quantized_keys_to_ignore_on_load
118-
else:
119-
model_class._keys_to_ignore_on_load_unexpected.extend(quantized_keys_to_ignore_on_load)
120-
missing_keys_to_ignore_on_load = [r"weight", r"bias"]
121-
if keys_to_ignore_on_load_missing is None:
122-
model_class._keys_to_ignore_on_load_missing = missing_keys_to_ignore_on_load
123-
else: # pragma: no cover
124-
model_class._keys_to_ignore_on_load_missing.extend(missing_keys_to_ignore_on_load)
125-
99+
# only show logs of error level, since keys_to_ignore_on_load_unexpected is not working without specific model_class
100+
transformers.logging.set_verbosity_error()
126101
if not os.path.isdir(model_name_or_path) and not os.path.isfile(model_name_or_path): # pragma: no cover
127102
from transformers.utils import cached_file
128103

@@ -169,9 +144,6 @@ def from_pretrained(cls, model_name_or_path: str, **kwargs) -> torch.nn.Module:
169144
**kwargs,
170145
)
171146

172-
model_class._keys_to_ignore_on_load_unexpected = keys_to_ignore_on_load_unexpected
173-
model_class._keys_to_ignore_on_load_missing = keys_to_ignore_on_load_missing
174-
175147
if not os.path.isdir(model_name_or_path) and not os.path.isfile(model_name_or_path): # pragma: no cover
176148
# pylint: disable=E0611
177149
from packaging.version import Version

0 commit comments

Comments
 (0)