From 0c552a718f9c5d6fdc09b1e7913e707d0e816ea7 Mon Sep 17 00:00:00 2001 From: DesmonDay <908660116@qq.com> Date: Fri, 7 Jun 2024 17:58:52 +0800 Subject: [PATCH] Do not save moe_group --- paddlenlp/transformers/configuration_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddlenlp/transformers/configuration_utils.py b/paddlenlp/transformers/configuration_utils.py index 4bda24695a48..b3e255f30535 100644 --- a/paddlenlp/transformers/configuration_utils.py +++ b/paddlenlp/transformers/configuration_utils.py @@ -903,6 +903,8 @@ def to_dict(self) -> Dict[str, Any]: output["model_type"] = self.__class__.model_type if "_auto_class" in output: del output["_auto_class"] + if "moe_group" in output: + del output["moe_group"] output["quantization_config"] = self.quantization_config.to_dict()