Add free_dimension_overrides

#4
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -13,6 +13,14 @@
13
  "num_hidden_layers": 5,
14
  "patch_size": 2,
15
  "transformers_version": "4.11.0.dev0",
 
 
 
 
 
 
 
 
16
  "vocab_size": {},
17
  "architectures": [
18
  "ViTForImageClassification"
 
13
  "num_hidden_layers": 5,
14
  "patch_size": 2,
15
  "transformers_version": "4.11.0.dev0",
16
+ "transformers.js_config": {
17
+ "free_dimension_overrides": {
18
+ "batch_size": 1,
19
+ "num_channels": 3,
20
+ "height": 30,
21
+ "width": 30
22
+ }
23
+ },
24
  "vocab_size": {},
25
  "architectures": [
26
  "ViTForImageClassification"