init branch

Chengqian-Zhang · Aug 28, 2024 · 390223d · 390223d
1 parent 15cd17a
commit 390223d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepmd/pt/infer/deep_eval.py b/deepmd/pt/infer/deep_eval.py
@@ -122,7 +122,8 @@ def __init__(
                         ] = state_dict[item].clone()
                 state_dict = state_dict_head
             model = get_model(self.input_param).to(DEVICE)
-            model = torch.jit.script(model)
+
+            #model = torch.jit.script(model)
             self.dp = ModelWrapper(model)
             self.dp.load_state_dict(state_dict)
         elif str(self.model_path).endswith(".pth"):