Update difpoint/src/models/predictor.py

difpoint/src/models/predictor.py

@@ -254,19 +254,20 @@ class OnnxRuntimePredictorSingleton(OnnxRuntimePredictor):
     """
     单例模式，防止模型被加载多次
     """
-    _instance_lock = threading.Lock()
-    _instance = {}
+    #_instance_lock = threading.Lock()
+    #_instance = {}
 
     def __new__(cls, *args, **kwargs):
         model_path = kwargs.get("model_path", "")  # 用模型路径区分是否是一样的实例
         assert os.path.exists(model_path), "model path must exist!"
         # 单例模式，避免重复加载模型
-        with OnnxRuntimePredictorSingleton._instance_lock:
-            if model_path not in OnnxRuntimePredictorSingleton._instance or \
-                    OnnxRuntimePredictorSingleton._instance[model_path].onnx_model is None:
-                OnnxRuntimePredictorSingleton._instance[model_path] = OnnxRuntimePredictor(**kwargs)
+        #with OnnxRuntimePredictorSingleton._instance_lock:
+            #if model_path not in OnnxRuntimePredictorSingleton._instance or \
+                    #OnnxRuntimePredictorSingleton._instance[model_path].onnx_model is None:
+                #OnnxRuntimePredictorSingleton._instance[model_path] = OnnxRuntimePredictor(**kwargs)
 
-        return OnnxRuntimePredictorSingleton._instance[model_path]
+        #return OnnxRuntimePredictorSingleton._instance[model_path]
+        return OnnxRuntimePredictor(**kwargs)
 
 
 def get_predictor(**kwargs):