Skip to content

Commit e019e12

Browse files
Update deepmd/pd/infer/deep_eval.py
Co-authored-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu> Signed-off-by: HydrogenSulfate <490868991@qq.com>
1 parent 37d7271 commit e019e12

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/pd/infer/deep_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def __init__(
149149
self.dp.eval()
150150
self.static_model = False
151151
elif str(self.model_path).endswith(".json"):
152-
self.dp = paddle.jit.load(self.model_path.split(".json")[0])
152+
self.dp = paddle.jit.load(self.model_path[:-5])
153153
self.rcut = self.dp.get_buffer_rcut().item()
154154
self.type_map: list[str] = "".join(
155155
[chr(x) for x in self.dp.get_buffer_type_map().numpy()]

0 commit comments

Comments
 (0)