@@ -96,33 +96,8 @@ def from_pretrained(cls, model_name_or_path: str, **kwargs) -> torch.nn.Module:
96
96
)
97
97
return model
98
98
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 ()
126
101
if not os .path .isdir (model_name_or_path ) and not os .path .isfile (model_name_or_path ): # pragma: no cover
127
102
from transformers .utils import cached_file
128
103
@@ -169,9 +144,6 @@ def from_pretrained(cls, model_name_or_path: str, **kwargs) -> torch.nn.Module:
169
144
** kwargs ,
170
145
)
171
146
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
-
175
147
if not os .path .isdir (model_name_or_path ) and not os .path .isfile (model_name_or_path ): # pragma: no cover
176
148
# pylint: disable=E0611
177
149
from packaging .version import Version
0 commit comments