mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-04 05:10:06 +06:00
parent
9e730689c3
commit
9c5b1319d0
@ -845,6 +845,10 @@ class PretrainedConfig(PushToHubMixin):
|
|||||||
|
|
||||||
self._remove_keys_not_serialized(serializable_config_dict)
|
self._remove_keys_not_serialized(serializable_config_dict)
|
||||||
|
|
||||||
|
# Key removed only in diff dict
|
||||||
|
if "_name_or_path" in serializable_config_dict:
|
||||||
|
del serializable_config_dict["_name_or_path"]
|
||||||
|
|
||||||
if hasattr(self, "quantization_config"):
|
if hasattr(self, "quantization_config"):
|
||||||
serializable_config_dict["quantization_config"] = (
|
serializable_config_dict["quantization_config"] = (
|
||||||
self.quantization_config.to_dict()
|
self.quantization_config.to_dict()
|
||||||
@ -1005,8 +1009,6 @@ class PretrainedConfig(PushToHubMixin):
|
|||||||
# Do not serialize `base_model_pp_plan` for now
|
# Do not serialize `base_model_pp_plan` for now
|
||||||
if "base_model_pp_plan" in d:
|
if "base_model_pp_plan" in d:
|
||||||
del d["base_model_pp_plan"]
|
del d["base_model_pp_plan"]
|
||||||
if "_name_or_path" in d:
|
|
||||||
del d["_name_or_path"]
|
|
||||||
for value in d.values():
|
for value in d.values():
|
||||||
if isinstance(value, dict):
|
if isinstance(value, dict):
|
||||||
self._remove_keys_not_serialized(value)
|
self._remove_keys_not_serialized(value)
|
||||||
|
Loading…
Reference in New Issue
Block a user