fix uninitilized type_embedding

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 7, 2024 · 0fcd127 · 0fcd127
1 parent 9605e9a
commit 0fcd127
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepmd/tf/model/model.py b/deepmd/tf/model/model.py
@@ -812,6 +812,8 @@ def deserialize(cls, data: dict, suffix: str = "") -> "Descriptor":
         if descriptor.explicit_ntypes:
             type_embedding = descriptor.type_embedding
             fitting.dim_descrpt -= type_embedding.neuron[-1]
+        else:
+            type_embedding = None
         # BEGINE not supported keys
         data.pop("atom_exclude_types")
         data.pop("pair_exclude_types")