BUAADreamer hiyouga commited on
Commit
8e590fd
1 Parent(s): 55b19eb

Update config.json (#1)

Browse files

- Update config.json (1ae09c8926b6c0b628cef2498c75c994a4cc3fa0)


Co-authored-by: Yaowei Zheng <hiyouga@users.noreply.huggingface.co>

Files changed (1) hide show
  1. config.json +3 -8
config.json CHANGED
@@ -1,16 +1,13 @@
1
  {
2
- "_name_or_path": "",
3
  "architectures": [
4
  "LlavaForConditionalGeneration"
5
  ],
6
- "hidden_size": 4096,
7
  "ignore_index": -100,
8
  "image_token_index": 64000,
9
  "model_type": "llava",
10
  "pad_token_id": 64001,
11
  "projector_hidden_act": "gelu",
12
  "text_config": {
13
- "_name_or_path": "checkpoints/01-ai/Yi-6B-Chat",
14
  "architectures": [
15
  "LlamaForCausalLM"
16
  ],
@@ -19,14 +16,13 @@
19
  "num_key_value_heads": 4,
20
  "rms_norm_eps": 1e-05,
21
  "rope_theta": 5000000.0,
22
- "torch_dtype": "bfloat16",
23
  "vocab_size": 64064
24
  },
 
25
  "torch_dtype": "bfloat16",
26
  "transformers_version": "4.40.1",
27
  "use_cache": true,
28
  "vision_config": {
29
- "_name_or_path": "checkpoints/01-ai/Yi-VL-6B/vit/clip-vit-H-14-laion2B-s32B-b79K-yi-vl-6B-448",
30
  "architectures": [
31
  "CLIPVisionModel"
32
  ],
@@ -39,9 +35,8 @@
39
  "num_attention_heads": 16,
40
  "num_hidden_layers": 32,
41
  "patch_size": 14,
42
- "projection_dim": 1024,
43
- "torch_dtype": "float32"
44
  },
45
  "vision_feature_layer": -2,
46
  "vision_feature_select_strategy": "default"
47
- }
 
1
  {
 
2
  "architectures": [
3
  "LlavaForConditionalGeneration"
4
  ],
 
5
  "ignore_index": -100,
6
  "image_token_index": 64000,
7
  "model_type": "llava",
8
  "pad_token_id": 64001,
9
  "projector_hidden_act": "gelu",
10
  "text_config": {
 
11
  "architectures": [
12
  "LlamaForCausalLM"
13
  ],
 
16
  "num_key_value_heads": 4,
17
  "rms_norm_eps": 1e-05,
18
  "rope_theta": 5000000.0,
 
19
  "vocab_size": 64064
20
  },
21
+ "tie_word_embeddings": false,
22
  "torch_dtype": "bfloat16",
23
  "transformers_version": "4.40.1",
24
  "use_cache": true,
25
  "vision_config": {
 
26
  "architectures": [
27
  "CLIPVisionModel"
28
  ],
 
35
  "num_attention_heads": 16,
36
  "num_hidden_layers": 32,
37
  "patch_size": 14,
38
+ "projection_dim": 1024
 
39
  },
40
  "vision_feature_layer": -2,
41
  "vision_feature_select_strategy": "default"
42
+ }