lewtun HF Staff commited on
Commit
1bc8b36
·
1 Parent(s): 19bf21c

Fix inference

Browse files
Files changed (2) hide show
  1. config.json +6 -6
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -39,9 +39,9 @@
39
  ],
40
  "drop_path_rate": 0,
41
  "hidden_sizes": [
42
- 6,
43
- 12,
44
- 18
45
  ],
46
  "id2label": {
47
  "0": "tench, Tinca tinca",
@@ -2061,9 +2061,9 @@
2061
  ],
2062
  "model_type": "levit",
2063
  "num_attention_heads": [
2064
- 2,
2065
- 3,
2066
- 4
2067
  ],
2068
  "num_channels": 3,
2069
  "padding": 1,
 
39
  ],
40
  "drop_path_rate": 0,
41
  "hidden_sizes": [
42
+ 16,
43
+ 32,
44
+ 48
45
  ],
46
  "id2label": {
47
  "0": "tench, Tinca tinca",
 
2061
  ],
2062
  "model_type": "levit",
2063
  "num_attention_heads": [
2064
+ 4,
2065
+ 6,
2066
+ 8
2067
  ],
2068
  "num_channels": 3,
2069
  "padding": 1,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e4740a5527f81a1a60782d0b707302010fbe721a2814e862506e144125f96c89
3
- size 1426671
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7f25db703ea8ea8b69b64b2cd30d32e2f0b1b3214ff557bff596e6c37ca327c
3
+ size 2846895