Spaces:
Running
on
Zero
Running
on
Zero
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
|