From 617a2584323cee7f957c270c197688d60b74ad4c Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Mon, 2 Dec 2024 14:23:29 +0800 Subject: [PATCH] update __init__ --- deepmd/pd/model/descriptor/__init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/deepmd/pd/model/descriptor/__init__.py b/deepmd/pd/model/descriptor/__init__.py index 8af9582527..cee9dbf226 100644 --- a/deepmd/pd/model/descriptor/__init__.py +++ b/deepmd/pd/model/descriptor/__init__.py @@ -22,6 +22,10 @@ DescrptBlockSeA, DescrptSeA, ) +from .se_t_tebd import ( + DescrptBlockSeTTebd, + DescrptSeTTebd, +) __all__ = [ "BaseDescriptor", @@ -29,8 +33,10 @@ "DescrptBlockRepformers", "DescrptBlockSeA", "DescrptBlockSeAtten", + "DescrptBlockSeTTebd", "DescrptDPA1", "DescrptDPA2", "DescrptSeA", + "DescrptSeTTebd", "prod_env_mat", ]