From 390223d5c2d600e7858c5565c103186110204feb Mon Sep 17 00:00:00 2001
From: Chengqian-Zhang <2000011006@stu.pku.edu.cn>
Date: Wed, 28 Aug 2024 04:44:46 +0000
Subject: [PATCH] init branch

---
 deepmd/pt/infer/deep_eval.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deepmd/pt/infer/deep_eval.py b/deepmd/pt/infer/deep_eval.py
index c9c9c995d0..66b5470b7d 100644
--- 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"):