louisabraham-h
commited on
fix vision_config
Browse filessee https://github.com/OSU-NLP-Group/UGround/issues/12
- config.json +9 -2
config.json
CHANGED
@@ -35,9 +35,16 @@
|
|
35 |
"use_sliding_window": false,
|
36 |
"video_token_id": 151656,
|
37 |
"vision_config": {
|
|
|
|
|
|
|
|
|
38 |
"in_chans": 3,
|
39 |
-
"
|
40 |
-
"
|
|
|
|
|
|
|
41 |
},
|
42 |
"vision_end_token_id": 151653,
|
43 |
"vision_start_token_id": 151652,
|
|
|
35 |
"use_sliding_window": false,
|
36 |
"video_token_id": 151656,
|
37 |
"vision_config": {
|
38 |
+
"depth": 32,
|
39 |
+
"embed_dim": 1280,
|
40 |
+
"mlp_ratio": 4,
|
41 |
+
"num_heads": 16,
|
42 |
"in_chans": 3,
|
43 |
+
"hidden_size": 3584,
|
44 |
+
"patch_size": 14,
|
45 |
+
"spatial_merge_size": 2,
|
46 |
+
"spatial_patch_size": 14,
|
47 |
+
"temporal_patch_size": 2
|
48 |
},
|
49 |
"vision_end_token_id": 151653,
|
50 |
"vision_start_token_id": 151652,
|