From a650fddfb654e21cb826cd900c22f71052336727 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Mon, 2 Oct 2023 01:00:34 -0400
Subject: [PATCH] fix for MultiModel

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 deepmd/model/multi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepmd/model/multi.py b/deepmd/model/multi.py
index b0aa11a109..ec932b5838 100644
--- 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,
             )