fix DescrptSeTTebd

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
iProzd · Nov 21, 2024 · 6ec5f6f · 6ec5f6f
1 parent 4f4cbf5
commit 6ec5f6f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/dpmodel/descriptor/se_t_tebd.py b/deepmd/dpmodel/descriptor/se_t_tebd.py
@@ -346,6 +346,7 @@ def call(
             atype_ext,
             atype_embd_ext,
             mapping=None,
+            type_embedding=type_embedding,
         )
         # nf x nloc x (ng + tebd_dim)
         if self.concat_output_tebd: