fix for MultiModel

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Oct 2, 2023 · a650fdd · a650fdd
1 parent 2b90ebe
commit a650fdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/model/multi.py b/deepmd/model/multi.py
@@ -122,7 +122,7 @@ def __init__(
         else:
             self.descrpt = Descriptor(
                 **descriptor,
-                ntypes=len(type_map),
+                ntypes=len(self.type_map),
                 multi_task=True,
                 spin=self.spin,
             )