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 |
}
|