Don't save moe_group (PaddlePaddle#8570)

DesmonDay · Jun 7, 2024 · 5b027c8 · 5b027c8
1 parent c628f12
commit 5b027c8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()