ibelem commited on
Commit
e8b21c9
·
verified ·
1 Parent(s): 9c4d578

Add free_dimension_overrides in config.json

Browse files

Based on https://github.com/huggingface/transformers.js/blob/main/src/models.js#L239 , WebNN requires free_dimension_overrides to be set in config.json as a field within "transformers.js_config".

Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -40,5 +40,13 @@
40
  "patch_size": 2,
41
  "qkv_bias": true,
42
  "transformers_version": "4.48.2",
 
 
 
 
 
 
 
 
43
  "use_mid_position_embeddings": true
44
  }
 
40
  "patch_size": 2,
41
  "qkv_bias": true,
42
  "transformers_version": "4.48.2",
43
+ "transformers.js_config": {
44
+ "free_dimension_overrides": {
45
+ "batch_size": 1,
46
+ "num_channels": 3,
47
+ "height": 30,
48
+ "width": 30
49
+ }
50
+ },
51
  "use_mid_position_embeddings": true
52
  }