init for dpmodel make_model

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Mar 12, 2024 · 72fbcfd · 72fbcfd
1 parent 0eb7847
commit 72fbcfd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepmd/dpmodel/model/make_model.py b/deepmd/dpmodel/model/make_model.py
@@ -73,6 +73,7 @@ def __init__(
             atomic_model_: Optional[T_AtomicModel] = None,
             **kwargs,
         ):
+            BaseModel.__init__(self)
             if atomic_model_ is not None:
                 self.atomic_model: T_AtomicModel = atomic_model_
             else: