xyzhang626 commited on
Commit
fb244de
1 Parent(s): cc995f1

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": 1024,
@@ -16,7 +17,7 @@
16
  "intermediate_size": 4096,
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": 16,
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": 1024,
 
17
  "intermediate_size": 4096,
18
  "layer_norm_eps": 1e-12,
19
  "mask_ratio": 0.75,
20
+ "model_type": "vit",
21
  "norm_pix_loss": false,
22
  "num_attention_heads": 16,
23
  "num_channels": 3,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bf88b5aaf29f2dd648449cb15da6eebc094231c28518e100bf806cda1e4727a2
3
  size 1217695704
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e71da923ae715d82ec99e2e1c0d9da5aeaddc7dec44bd1e757a5af42bbfbc52b
3
  size 1217695704