Skip to content

Commit 5b027c8

Browse files
authored
Don't save moe_group (#8570)
1 parent c628f12 commit 5b027c8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddlenlp/transformers/configuration_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -903,6 +903,8 @@ def to_dict(self) -> Dict[str, Any]:
903903
output["model_type"] = self.__class__.model_type
904904
if "_auto_class" in output:
905905
del output["_auto_class"]
906+
if "moe_group" in output:
907+
del output["moe_group"]
906908

907909
output["quantization_config"] = self.quantization_config.to_dict()
908910

0 commit comments

Comments
 (0)