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
|