DegMaTsu commited on
Commit
7f8ebfb
·
verified ·
1 Parent(s): 0b1b1be

Upload reactor_utils.py

Browse files
custom_nodes/comfyui-reactor-node/reactor_utils.py CHANGED
@@ -188,6 +188,11 @@ def set_ort_session(model_path, providers) -> Any:
188
  onnxruntime.set_default_logger_severity(3)
189
  ORT_SESSION = onnxruntime.InferenceSession(model_path, providers=providers)
190
  return ORT_SESSION
 
 
 
 
 
191
 
192
  def clear_ort_session() -> None:
193
  global ORT_SESSION
 
188
  onnxruntime.set_default_logger_severity(3)
189
  ORT_SESSION = onnxruntime.InferenceSession(model_path, providers=providers)
190
  return ORT_SESSION
191
+
192
+ def set_ort_session(model_path, providers=None):
193
+ if not model_path or not os.path.isfile(model_path):
194
+ raise FileNotFoundError(f"ONNX-модель не найдена: {model_path}")
195
+ return onnxruntime.InferenceSession(model_path, providers=providers or ["CPUExecutionProvider"])
196
 
197
  def clear_ort_session() -> None:
198
  global ORT_SESSION