Skip to content

Commit 7b7cdfa

Browse files
committed
fix dpmodel repflow serialization
1 parent df875cf commit 7b7cdfa

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

deepmd/dpmodel/descriptor/repflows.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -585,6 +585,7 @@ def serialize(self):
585585
"repflow_layers": [layer.serialize() for layer in self.layers],
586586
"env_mat_edge": self.env_mat_edge.serialize(),
587587
"env_mat_angle": self.env_mat_angle.serialize(),
588+
"use_ext_ebd": self.use_ext_ebd,
588589
"@variables": {
589590
"davg": to_numpy_array(self["davg"]),
590591
"dstd": to_numpy_array(self["dstd"]),

0 commit comments

Comments
 (0)