fix(pt): fix model_def_script

In the current code, model_def_script is modified, causing incompatibility with other backends.
deepmodeling · Apr 14, 2024 · 3b230c2 · 3b230c2
1 parent 3eb3d51
commit 3b230c2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
@@ -138,6 +138,7 @@ def get_zbl_model(model_params):
 
 
 def get_standard_model(model_params):
+    model_params_old = model_params
     model_params = copy.deepcopy(model_params)
     ntypes = len(model_params["type_map"])
     # descriptor
@@ -167,7 +168,7 @@ def get_standard_model(model_params):
         atom_exclude_types=atom_exclude_types,
         pair_exclude_types=pair_exclude_types,
     )
-    model.model_def_script = json.dumps(model_params)
+    model.model_def_script = json.dumps(model_params_old)
     return model