| #!/usr/bin/env python3 | |
| """Runtime configuration management.""" | |
| from onnx import helper | |
| def get_providers(device='auto'): | |
| """Get ONNX Runtime execution providers based on device.""" | |
| if device == 'cuda': | |
| return ['CUDAExecutionProvider', 'CPUExecutionProvider'] | |
| return ['CPUExecutionProvider'] | |
| def make_opset(version=17): | |
| """Create ONNX opset identifier.""" | |
| return [helper.make_opsetid("", version)] | |