xyzhang626 commited on
Commit
d5772e1
1 Parent(s): cd050b7

Upload model

Browse files
Files changed (2) hide show
  1. config.json +2 -1
  2. model.safetensors +1 -1
config.json CHANGED
@@ -8,6 +8,7 @@
8
  "decoder_intermediate_size": 2048,
9
  "decoder_num_attention_heads": 16,
10
  "decoder_num_hidden_layers": 8,
 
11
  "hidden_act": "gelu",
12
  "hidden_dropout_prob": 0.0,
13
  "hidden_size": 768,
@@ -16,7 +17,7 @@
16
  "intermediate_size": 3072,
17
  "layer_norm_eps": 1e-12,
18
  "mask_ratio": 0.75,
19
- "model_type": "vit_mae",
20
  "norm_pix_loss": false,
21
  "num_attention_heads": 12,
22
  "num_channels": 3,
 
8
  "decoder_intermediate_size": 2048,
9
  "decoder_num_attention_heads": 16,
10
  "decoder_num_hidden_layers": 8,
11
+ "encoder_stride": 16,
12
  "hidden_act": "gelu",
13
  "hidden_dropout_prob": 0.0,
14
  "hidden_size": 768,
 
17
  "intermediate_size": 3072,
18
  "layer_norm_eps": 1e-12,
19
  "mask_ratio": 0.75,
20
+ "model_type": "vit",
21
  "norm_pix_loss": false,
22
  "num_attention_heads": 12,
23
  "num_channels": 3,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5b88f0a9dcce3999d111bda1909e095db58ca51742ef86aa9c48bf105f95093f
3
  size 345763744
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa7d788e37a350f1054e5260d2452ebf1b7adafa027332a68a61af27111d2324
3
  size 345763744