Skip to content

Commit 9025172

Browse files
committed
Update deep_eval.py
1 parent 20e7d9b commit 9025172

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

deepmd/pt/infer/deep_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def __init__(
130130
] = state_dict[item].clone()
131131
state_dict = state_dict_head
132132
model = get_model(self.input_param).to(DEVICE)
133-
model = torch.jit.script(model)
133+
# model = torch.jit.script(model)
134134
self.dp = ModelWrapper(model)
135135
self.dp.load_state_dict(state_dict)
136136
elif str(self.model_path).endswith(".pth"):

0 commit comments

Comments
 (0)