Testing device-level configs
Browse filesSee https://github.com/huggingface/transformers.js/pull/1276
- config.json +11 -1
config.json
CHANGED
@@ -1,6 +1,16 @@
|
|
1 |
{
|
2 |
"model_type": "modnet",
|
3 |
"transformers.js_config": {
|
4 |
-
"dtype": "fp32"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
}
|
6 |
}
|
|
|
1 |
{
|
2 |
"model_type": "modnet",
|
3 |
"transformers.js_config": {
|
4 |
+
"dtype": "fp32",
|
5 |
+
"device_config": {
|
6 |
+
"webnn": {
|
7 |
+
"free_dimension_overrides": {
|
8 |
+
"batch_size": 1,
|
9 |
+
"num_channels": 3,
|
10 |
+
"height": 256,
|
11 |
+
"width": 256
|
12 |
+
}
|
13 |
+
}
|
14 |
+
}
|
15 |
}
|
16 |
}
|