Update __init__.py

iProzd · Feb 27, 2024 · c2f306c · c2f306c
1 parent 796d870
commit c2f306c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/model/model/__init__.py b/deepmd/pt/model/model/__init__.py
@@ -135,7 +135,7 @@ def get_spin_model(model_params):
     model_params["fitting_net"]["exclude_types"] = atom_exclude_types
     # descriptor
     model_params["descriptor"]["ntypes"] = ntypes
-    descriptor = Descriptor(**model_params["descriptor"])
+    descriptor = BaseDescriptor(**model_params["descriptor"])
     # fitting
     fitting_net = model_params.get("fitting_net", None)
     fitting_net["type"] = fitting_net.get("type", "ener")