Xenova HF Staff commited on
Commit
252d733
·
verified ·
1 Parent(s): e71ff16

Testing device-level configs

Browse files

See https://github.com/huggingface/transformers.js/pull/1276

Files changed (1) hide show
  1. 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
  }