fix: zbl rcut, sel

deepmodeling · Nov 11, 2024 · 050edd4 · 050edd4
1 parent dcbf607
commit 050edd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
@@ -180,8 +180,8 @@ def get_zbl_model(model_params):
     filepath = model_params["use_srtab"]
     pt_model = PairTabAtomicModel(
         filepath,
-        model_params["descriptor"]["rcut"],
-        model_params["descriptor"]["sel"],
+        descriptor.get_rcut(),
+        descriptor.get_sel(),
         type_map=model_params["type_map"],
     )