There was an error while loading. Please reload this page.
1 parent 37d7271 commit e019e12Copy full SHA for e019e12
deepmd/pd/infer/deep_eval.py
@@ -149,7 +149,7 @@ def __init__(
149
self.dp.eval()
150
self.static_model = False
151
elif str(self.model_path).endswith(".json"):
152
- self.dp = paddle.jit.load(self.model_path.split(".json")[0])
+ self.dp = paddle.jit.load(self.model_path[:-5])
153
self.rcut = self.dp.get_buffer_rcut().item()
154
self.type_map: list[str] = "".join(
155
[chr(x) for x in self.dp.get_buffer_type_map().numpy()]
0 commit comments