Propose CUDA and CoreML provider support
#1
by Neelesh-Babu - opened
- config.yaml +6 -0
config.yaml
CHANGED
|
@@ -35,6 +35,8 @@ artifacts:
|
|
| 35 |
role: asset
|
| 36 |
runtime:
|
| 37 |
execution_providers:
|
|
|
|
|
|
|
| 38 |
- CPUExecutionProvider
|
| 39 |
fusion: normalized_mean
|
| 40 |
components:
|
|
@@ -57,6 +59,8 @@ runtime:
|
|
| 57 |
- batch
|
| 58 |
- 512
|
| 59 |
execution_providers:
|
|
|
|
|
|
|
| 60 |
- CPUExecutionProvider
|
| 61 |
preparation:
|
| 62 |
width: 224
|
|
@@ -123,6 +127,8 @@ runtime:
|
|
| 123 |
- batch
|
| 124 |
- 512
|
| 125 |
execution_providers:
|
|
|
|
|
|
|
| 126 |
- CPUExecutionProvider
|
| 127 |
preparation:
|
| 128 |
tokenizer_path: tokenizer
|
|
|
|
| 35 |
role: asset
|
| 36 |
runtime:
|
| 37 |
execution_providers:
|
| 38 |
+
- CUDAExecutionProvider
|
| 39 |
+
- CoreMLExecutionProvider
|
| 40 |
- CPUExecutionProvider
|
| 41 |
fusion: normalized_mean
|
| 42 |
components:
|
|
|
|
| 59 |
- batch
|
| 60 |
- 512
|
| 61 |
execution_providers:
|
| 62 |
+
- CUDAExecutionProvider
|
| 63 |
+
- CoreMLExecutionProvider
|
| 64 |
- CPUExecutionProvider
|
| 65 |
preparation:
|
| 66 |
width: 224
|
|
|
|
| 127 |
- batch
|
| 128 |
- 512
|
| 129 |
execution_providers:
|
| 130 |
+
- CUDAExecutionProvider
|
| 131 |
+
- CoreMLExecutionProvider
|
| 132 |
- CPUExecutionProvider
|
| 133 |
preparation:
|
| 134 |
tokenizer_path: tokenizer
|