cheryyunl commited on
Commit
f92d45b
·
verified ·
1 Parent(s): 70f421d

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
added_tokens.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</pointcloud>": 151666,
3
+ "</tool_call>": 151658,
4
+ "<layer_sep>": 151669,
5
+ "<point_patch>": 151667,
6
+ "<pointcloud>": 151665,
7
+ "<row_sep>": 151668,
8
+ "<tool_call>": 151657,
9
+ "<|box_end|>": 151649,
10
+ "<|box_start|>": 151648,
11
+ "<|endoftext|>": 151643,
12
+ "<|file_sep|>": 151664,
13
+ "<|fim_middle|>": 151660,
14
+ "<|fim_pad|>": 151662,
15
+ "<|fim_prefix|>": 151659,
16
+ "<|fim_suffix|>": 151661,
17
+ "<|im_end|>": 151645,
18
+ "<|im_start|>": 151644,
19
+ "<|image_pad|>": 151655,
20
+ "<|object_ref_end|>": 151647,
21
+ "<|object_ref_start|>": 151646,
22
+ "<|quad_end|>": 151651,
23
+ "<|quad_start|>": 151650,
24
+ "<|repo_name|>": 151663,
25
+ "<|video_pad|>": 151656,
26
+ "<|vision_end|>": 151653,
27
+ "<|vision_pad|>": 151654,
28
+ "<|vision_start|>": 151652
29
+ }
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "MultimodalQwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 151643,
7
+ "eos_token_id": 151645,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 3584,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 18944,
12
+ "max_position_embeddings": 32768,
13
+ "max_window_layers": 28,
14
+ "model_type": "multimodal_qwen2",
15
+ "num_attention_heads": 28,
16
+ "num_hidden_layers": 28,
17
+ "num_key_value_heads": 4,
18
+ "point_patch_size": 512,
19
+ "rms_norm_eps": 1e-06,
20
+ "rope_scaling": null,
21
+ "rope_theta": 1000000.0,
22
+ "sliding_window": 131072,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "float32",
25
+ "transformers_version": "4.51.3",
26
+ "use_cache": true,
27
+ "use_sliding_window": false,
28
+ "vocab_size": 152064,
29
+ "auto_map": {
30
+ "AutoConfig": "modeling_lemon.MultimodalQwen2Config",
31
+ "AutoModelForCausalLM": "modeling_lemon.MultimodalQwen2ForCausalLM"
32
+ }
33
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 151643,
4
+ "eos_token_id": 151645,
5
+ "transformers_version": "4.51.3"
6
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model-00001-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f66cc862ed84e8f6a78a0f27c0f7fb08be2d7005fbde3e160ccdb82845c48ce2
3
+ size 4976687216
model-00002-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0edb205482e80b5d03e63693e387d459b0cb9c97948cde971ff025ba98f1396d
3
+ size 4778622352
model-00003-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2222b40c6fea7def1d5911b059ab36bf97b9328e0251689f78a0a9ebd98cd18d
3
+ size 4932743960
model-00004-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfc2c86e34819a46908bfe3ad42f3986e7efdfc34f67e43f40f690f77c7b0672
3
+ size 4932743992
model-00005-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:891f86a74d6b1791b98b581ff94cbedb16bfdb78595b9697dc3fd2ecc5974ba2
3
+ size 4998852296
model-00006-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4be5dbaf63c63933069328ccee987d31303e79fadd834497d2543e127c730791
3
+ size 3684885392
model-00007-of-00007.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c550566f2a473b9db91b2cf032b3e2e4b2563ee8d2753e555203077aaf0d9349
3
+ size 2179989632
model.safetensors.index.json ADDED
@@ -0,0 +1,347 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 30484486144
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00007-of-00007.safetensors",
7
+ "model.embed_point_patch.weight": "model-00006-of-00007.safetensors",
8
+ "model.embed_tokens.weight": "model-00001-of-00007.safetensors",
9
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00007.safetensors",
10
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
11
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
12
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
13
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
14
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
15
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
16
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
17
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
18
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
19
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
20
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
21
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00007.safetensors",
22
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
23
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
24
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
25
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
26
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
27
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
28
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
29
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
30
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
31
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
32
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
33
+ "model.layers.10.input_layernorm.weight": "model-00003-of-00007.safetensors",
34
+ "model.layers.10.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
35
+ "model.layers.10.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
36
+ "model.layers.10.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
37
+ "model.layers.10.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
38
+ "model.layers.10.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
39
+ "model.layers.10.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
40
+ "model.layers.10.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
41
+ "model.layers.10.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
42
+ "model.layers.10.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
43
+ "model.layers.10.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
44
+ "model.layers.10.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
45
+ "model.layers.11.input_layernorm.weight": "model-00003-of-00007.safetensors",
46
+ "model.layers.11.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
47
+ "model.layers.11.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
48
+ "model.layers.11.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
49
+ "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
50
+ "model.layers.11.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
51
+ "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
52
+ "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
53
+ "model.layers.11.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
54
+ "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
55
+ "model.layers.11.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
56
+ "model.layers.11.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
57
+ "model.layers.12.input_layernorm.weight": "model-00003-of-00007.safetensors",
58
+ "model.layers.12.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
59
+ "model.layers.12.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
60
+ "model.layers.12.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
61
+ "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
62
+ "model.layers.12.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
63
+ "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
64
+ "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
65
+ "model.layers.12.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
66
+ "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
67
+ "model.layers.12.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
68
+ "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
69
+ "model.layers.13.input_layernorm.weight": "model-00004-of-00007.safetensors",
70
+ "model.layers.13.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
71
+ "model.layers.13.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
72
+ "model.layers.13.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
73
+ "model.layers.13.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
74
+ "model.layers.13.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
75
+ "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
76
+ "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
77
+ "model.layers.13.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
78
+ "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
79
+ "model.layers.13.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
80
+ "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
81
+ "model.layers.14.input_layernorm.weight": "model-00004-of-00007.safetensors",
82
+ "model.layers.14.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
83
+ "model.layers.14.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
84
+ "model.layers.14.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
85
+ "model.layers.14.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
86
+ "model.layers.14.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
87
+ "model.layers.14.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
88
+ "model.layers.14.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
89
+ "model.layers.14.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
90
+ "model.layers.14.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
91
+ "model.layers.14.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
92
+ "model.layers.14.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
93
+ "model.layers.15.input_layernorm.weight": "model-00004-of-00007.safetensors",
94
+ "model.layers.15.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
95
+ "model.layers.15.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
96
+ "model.layers.15.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
97
+ "model.layers.15.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
98
+ "model.layers.15.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
99
+ "model.layers.15.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
100
+ "model.layers.15.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
101
+ "model.layers.15.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
102
+ "model.layers.15.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
103
+ "model.layers.15.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
104
+ "model.layers.15.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
105
+ "model.layers.16.input_layernorm.weight": "model-00004-of-00007.safetensors",
106
+ "model.layers.16.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
107
+ "model.layers.16.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
108
+ "model.layers.16.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
109
+ "model.layers.16.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
110
+ "model.layers.16.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
111
+ "model.layers.16.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
112
+ "model.layers.16.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
113
+ "model.layers.16.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
114
+ "model.layers.16.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
115
+ "model.layers.16.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
116
+ "model.layers.16.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
117
+ "model.layers.17.input_layernorm.weight": "model-00004-of-00007.safetensors",
118
+ "model.layers.17.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
119
+ "model.layers.17.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
120
+ "model.layers.17.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
121
+ "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
122
+ "model.layers.17.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
123
+ "model.layers.17.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
124
+ "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
125
+ "model.layers.17.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
126
+ "model.layers.17.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
127
+ "model.layers.17.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
128
+ "model.layers.17.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
129
+ "model.layers.18.input_layernorm.weight": "model-00005-of-00007.safetensors",
130
+ "model.layers.18.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
131
+ "model.layers.18.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
132
+ "model.layers.18.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
133
+ "model.layers.18.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
134
+ "model.layers.18.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
135
+ "model.layers.18.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
136
+ "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
137
+ "model.layers.18.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
138
+ "model.layers.18.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
139
+ "model.layers.18.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
140
+ "model.layers.18.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
141
+ "model.layers.19.input_layernorm.weight": "model-00005-of-00007.safetensors",
142
+ "model.layers.19.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
143
+ "model.layers.19.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
144
+ "model.layers.19.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
145
+ "model.layers.19.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
146
+ "model.layers.19.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
147
+ "model.layers.19.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
148
+ "model.layers.19.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
149
+ "model.layers.19.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
150
+ "model.layers.19.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
151
+ "model.layers.19.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
152
+ "model.layers.19.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
153
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00007.safetensors",
154
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
155
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
156
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
157
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
158
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
159
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
160
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
161
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
162
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
163
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
164
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
165
+ "model.layers.20.input_layernorm.weight": "model-00005-of-00007.safetensors",
166
+ "model.layers.20.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
167
+ "model.layers.20.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
168
+ "model.layers.20.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
169
+ "model.layers.20.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
170
+ "model.layers.20.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
171
+ "model.layers.20.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
172
+ "model.layers.20.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
173
+ "model.layers.20.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
174
+ "model.layers.20.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
175
+ "model.layers.20.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
176
+ "model.layers.20.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
177
+ "model.layers.21.input_layernorm.weight": "model-00005-of-00007.safetensors",
178
+ "model.layers.21.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
179
+ "model.layers.21.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
180
+ "model.layers.21.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
181
+ "model.layers.21.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
182
+ "model.layers.21.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
183
+ "model.layers.21.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
184
+ "model.layers.21.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
185
+ "model.layers.21.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
186
+ "model.layers.21.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
187
+ "model.layers.21.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
188
+ "model.layers.21.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
189
+ "model.layers.22.input_layernorm.weight": "model-00005-of-00007.safetensors",
190
+ "model.layers.22.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
191
+ "model.layers.22.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
192
+ "model.layers.22.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
193
+ "model.layers.22.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
194
+ "model.layers.22.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
195
+ "model.layers.22.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
196
+ "model.layers.22.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
197
+ "model.layers.22.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
198
+ "model.layers.22.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
199
+ "model.layers.22.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
200
+ "model.layers.22.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
201
+ "model.layers.23.input_layernorm.weight": "model-00005-of-00007.safetensors",
202
+ "model.layers.23.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
203
+ "model.layers.23.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
204
+ "model.layers.23.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
205
+ "model.layers.23.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
206
+ "model.layers.23.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
207
+ "model.layers.23.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
208
+ "model.layers.23.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
209
+ "model.layers.23.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
210
+ "model.layers.23.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
211
+ "model.layers.23.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
212
+ "model.layers.23.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
213
+ "model.layers.24.input_layernorm.weight": "model-00006-of-00007.safetensors",
214
+ "model.layers.24.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
215
+ "model.layers.24.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
216
+ "model.layers.24.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
217
+ "model.layers.24.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
218
+ "model.layers.24.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
219
+ "model.layers.24.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
220
+ "model.layers.24.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
221
+ "model.layers.24.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
222
+ "model.layers.24.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
223
+ "model.layers.24.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
224
+ "model.layers.24.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
225
+ "model.layers.25.input_layernorm.weight": "model-00006-of-00007.safetensors",
226
+ "model.layers.25.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
227
+ "model.layers.25.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
228
+ "model.layers.25.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
229
+ "model.layers.25.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
230
+ "model.layers.25.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
231
+ "model.layers.25.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
232
+ "model.layers.25.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
233
+ "model.layers.25.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
234
+ "model.layers.25.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
235
+ "model.layers.25.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
236
+ "model.layers.25.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
237
+ "model.layers.26.input_layernorm.weight": "model-00006-of-00007.safetensors",
238
+ "model.layers.26.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
239
+ "model.layers.26.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
240
+ "model.layers.26.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
241
+ "model.layers.26.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
242
+ "model.layers.26.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
243
+ "model.layers.26.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
244
+ "model.layers.26.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
245
+ "model.layers.26.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
246
+ "model.layers.26.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
247
+ "model.layers.26.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
248
+ "model.layers.26.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
249
+ "model.layers.27.input_layernorm.weight": "model-00006-of-00007.safetensors",
250
+ "model.layers.27.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
251
+ "model.layers.27.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
252
+ "model.layers.27.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
253
+ "model.layers.27.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
254
+ "model.layers.27.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
255
+ "model.layers.27.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
256
+ "model.layers.27.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
257
+ "model.layers.27.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
258
+ "model.layers.27.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
259
+ "model.layers.27.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
260
+ "model.layers.27.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
261
+ "model.layers.3.input_layernorm.weight": "model-00002-of-00007.safetensors",
262
+ "model.layers.3.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
263
+ "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
264
+ "model.layers.3.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
265
+ "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
266
+ "model.layers.3.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
267
+ "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
268
+ "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
269
+ "model.layers.3.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
270
+ "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
271
+ "model.layers.3.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
272
+ "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
273
+ "model.layers.4.input_layernorm.weight": "model-00002-of-00007.safetensors",
274
+ "model.layers.4.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
275
+ "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
276
+ "model.layers.4.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
277
+ "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
278
+ "model.layers.4.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
279
+ "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
280
+ "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
281
+ "model.layers.4.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
282
+ "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
283
+ "model.layers.4.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
284
+ "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
285
+ "model.layers.5.input_layernorm.weight": "model-00002-of-00007.safetensors",
286
+ "model.layers.5.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
287
+ "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
288
+ "model.layers.5.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
289
+ "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
290
+ "model.layers.5.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
291
+ "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
292
+ "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
293
+ "model.layers.5.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
294
+ "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
295
+ "model.layers.5.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
296
+ "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
297
+ "model.layers.6.input_layernorm.weight": "model-00002-of-00007.safetensors",
298
+ "model.layers.6.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
299
+ "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
300
+ "model.layers.6.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
301
+ "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
302
+ "model.layers.6.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
303
+ "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
304
+ "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
305
+ "model.layers.6.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
306
+ "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
307
+ "model.layers.6.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
308
+ "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
309
+ "model.layers.7.input_layernorm.weight": "model-00002-of-00007.safetensors",
310
+ "model.layers.7.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
311
+ "model.layers.7.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
312
+ "model.layers.7.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
313
+ "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
314
+ "model.layers.7.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
315
+ "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
316
+ "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
317
+ "model.layers.7.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
318
+ "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
319
+ "model.layers.7.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
320
+ "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
321
+ "model.layers.8.input_layernorm.weight": "model-00003-of-00007.safetensors",
322
+ "model.layers.8.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
323
+ "model.layers.8.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
324
+ "model.layers.8.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
325
+ "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
326
+ "model.layers.8.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
327
+ "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
328
+ "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
329
+ "model.layers.8.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
330
+ "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
331
+ "model.layers.8.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
332
+ "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
333
+ "model.layers.9.input_layernorm.weight": "model-00003-of-00007.safetensors",
334
+ "model.layers.9.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
335
+ "model.layers.9.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
336
+ "model.layers.9.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
337
+ "model.layers.9.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
338
+ "model.layers.9.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
339
+ "model.layers.9.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
340
+ "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
341
+ "model.layers.9.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
342
+ "model.layers.9.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
343
+ "model.layers.9.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
344
+ "model.layers.9.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
345
+ "model.norm.weight": "model-00006-of-00007.safetensors"
346
+ }
347
+ }
modeling_lemon.py ADDED
@@ -0,0 +1,256 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional, List, Union, Tuple
2
+ import torch
3
+ import torch.nn as nn
4
+ from transformers import PretrainedConfig
5
+ from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast
6
+ from transformers.models.qwen2.modeling_qwen2 import (
7
+ Qwen2Config,
8
+ Qwen2PreTrainedModel,
9
+ Qwen2Model,
10
+ Qwen2ForCausalLM
11
+ )
12
+
13
+ class MultimodalQwen2Config(Qwen2Config):
14
+ """
15
+ Configuration class for MultimodalQwen2Model.
16
+ Extends Qwen2Config to include multimodal specific parameters.
17
+ """
18
+ model_type = "multimodal_qwen2"
19
+ def __init__(
20
+ self,
21
+ point_patch_size=512,
22
+ **kwargs
23
+ ):
24
+ super().__init__(**kwargs)
25
+ self.point_patch_size = point_patch_size
26
+
27
+
28
+ class MultimodalQwen2Model(Qwen2Model):
29
+ """
30
+ Multimodal Qwen2 model that supports point cloud data alongside text input.
31
+ """
32
+ config_class = MultimodalQwen2Config
33
+ def __init__(self, config: MultimodalQwen2Config):
34
+ super().__init__(config)
35
+
36
+ # Point cloud patch embedding layer
37
+ self.embed_point_patch = nn.Linear(
38
+ config.point_patch_size * 6, # 512 * 6
39
+ config.hidden_size,
40
+ bias=False,
41
+ dtype=torch.bfloat16
42
+ )
43
+
44
+ def forward(
45
+ self,
46
+ input_ids: torch.LongTensor = None,
47
+ attention_mask: Optional[torch.Tensor] = None,
48
+ point_patch_indices: torch.LongTensor = None, # (batch_size, seq_length), "-1" for text token
49
+ point_patches: torch.FloatTensor = None, # (n_patches, point_patch_size * 6)
50
+ position_ids: Optional[torch.LongTensor] = None,
51
+ past_key_values: Optional[List[torch.FloatTensor]] = None,
52
+ inputs_embeds: Optional[torch.FloatTensor] = None,
53
+ use_cache: Optional[bool] = None,
54
+ output_attentions: Optional[bool] = None,
55
+ output_hidden_states: Optional[bool] = None,
56
+ return_dict: Optional[bool] = None,
57
+ cache_position: Optional[torch.LongTensor] = None,
58
+ **kwargs
59
+ ) -> Union[Tuple, BaseModelOutputWithPast]:
60
+
61
+ if inputs_embeds is None:
62
+ inputs_embeds = self.embed_tokens(input_ids)
63
+
64
+ # Process point cloud patches
65
+ if point_patches is not None and point_patch_indices is not None:
66
+ # Validate input shape consistency
67
+ point_patches = point_patches.to(dtype=inputs_embeds.dtype, device=inputs_embeds.device)
68
+ if point_patch_indices.shape[1] != input_ids.shape[1]:
69
+ batch_size, target_len = input_ids.shape
70
+ current_len = point_patch_indices.shape[1]
71
+
72
+ if target_len > current_len:
73
+ padding = torch.full((batch_size, target_len - current_len), -1,
74
+ dtype=point_patch_indices.dtype,
75
+ device=point_patch_indices.device)
76
+ point_patch_indices = torch.cat([point_patch_indices, padding], dim=1)
77
+ else:
78
+ raise ValueError(
79
+ f"Cannot truncate point_patch_indices from {current_len} to {target_len}. "
80
+ f"This might lose point cloud patch mappings. "
81
+ )
82
+
83
+ # Embed point cloud patches
84
+ point_embeds = self.embed_point_patch(point_patches)
85
+
86
+ # Add dummy token for text (index -1)
87
+ point_embeds = torch.cat([
88
+ point_embeds,
89
+ torch.zeros(1, self.config.hidden_size).to(point_embeds)
90
+ ]) # (n_patches + 1, hidden_size)
91
+
92
+ # Arrange embeddings according to point_patch_indices
93
+ point_embeds = point_embeds[point_patch_indices] # (batch_size, seq_length, hidden_size)
94
+
95
+ # Merge point cloud embeddings with text embeddings
96
+ inputs_embeds = inputs_embeds + point_embeds
97
+
98
+ # Call parent's forward method to handle the rest
99
+ return super().forward(
100
+ input_ids=None,
101
+ attention_mask=attention_mask,
102
+ position_ids=position_ids,
103
+ past_key_values=past_key_values,
104
+ inputs_embeds=inputs_embeds,
105
+ use_cache=use_cache,
106
+ output_attentions=output_attentions,
107
+ output_hidden_states=output_hidden_states,
108
+ return_dict=return_dict,
109
+ cache_position=cache_position,
110
+ **kwargs
111
+ )
112
+
113
+
114
+ class MultimodalQwen2ForCausalLM(Qwen2ForCausalLM):
115
+ """
116
+ Multimodal Qwen2 model for causal language modeling.
117
+ Supports both text and point cloud inputs.
118
+ """
119
+ config_class = MultimodalQwen2Config
120
+ def __init__(self, config: MultimodalQwen2Config):
121
+ super().__init__(config)
122
+ # Replace base model with multimodal model
123
+ self.model = MultimodalQwen2Model(config)
124
+ # Re-apply initialization
125
+ self.post_init()
126
+
127
+ def forward(
128
+ self,
129
+ input_ids: torch.LongTensor = None,
130
+ attention_mask: Optional[torch.Tensor] = None,
131
+ point_patch_indices: torch.LongTensor = None,
132
+ point_patches: torch.FloatTensor = None,
133
+ position_ids: Optional[torch.LongTensor] = None,
134
+ past_key_values: Optional[Union[List[torch.FloatTensor]]] = None,
135
+ inputs_embeds: Optional[torch.FloatTensor] = None,
136
+ labels: Optional[torch.LongTensor] = None,
137
+ use_cache: Optional[bool] = None,
138
+ output_attentions: Optional[bool] = None,
139
+ output_hidden_states: Optional[bool] = None,
140
+ return_dict: Optional[bool] = None,
141
+ cache_position: Optional[torch.LongTensor] = None,
142
+ logits_to_keep: Union[int, torch.Tensor] = 0,
143
+ **kwargs
144
+ ) -> Union[Tuple, CausalLMOutputWithPast]:
145
+
146
+ output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
147
+ output_hidden_states = (
148
+ output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
149
+ )
150
+ return_dict = return_dict if return_dict is not None else self.config.use_return_dict
151
+
152
+ # Call multimodal model
153
+ outputs = self.model(
154
+ input_ids=input_ids,
155
+ attention_mask=attention_mask,
156
+ point_patch_indices=point_patch_indices,
157
+ point_patches=point_patches,
158
+ position_ids=position_ids,
159
+ past_key_values=past_key_values,
160
+ inputs_embeds=inputs_embeds,
161
+ use_cache=use_cache,
162
+ output_attentions=output_attentions,
163
+ output_hidden_states=output_hidden_states,
164
+ return_dict=return_dict,
165
+ cache_position=cache_position,
166
+ **kwargs
167
+ )
168
+
169
+ hidden_states = outputs[0]
170
+ # Only compute necessary logits based on logits_to_keep
171
+ slice_indices = slice(-logits_to_keep, None) if isinstance(logits_to_keep, int) else logits_to_keep
172
+ logits = self.lm_head(hidden_states[:, slice_indices, :])
173
+
174
+ loss = None
175
+ if labels is not None:
176
+ # Use Qwen's loss calculation function
177
+ loss = self.loss_function(logits=logits, labels=labels, vocab_size=self.config.vocab_size, **kwargs)
178
+
179
+ if not return_dict:
180
+ output = (logits,) + outputs[1:]
181
+ return (loss,) + output if loss is not None else output
182
+
183
+ return CausalLMOutputWithPast(
184
+ loss=loss,
185
+ logits=logits,
186
+ past_key_values=outputs.past_key_values,
187
+ hidden_states=outputs.hidden_states,
188
+ attentions=outputs.attentions,
189
+ )
190
+ def _validate_model_kwargs(self, model_kwargs):
191
+ filtered_kwargs = {k: v for k, v in model_kwargs.items()
192
+ if k not in ['point_patch_indices', 'point_patches']}
193
+ super()._validate_model_kwargs(filtered_kwargs)
194
+
195
+ def prepare_inputs_for_generation(
196
+ self,
197
+ input_ids,
198
+ past_key_values=None,
199
+ attention_mask=None,
200
+ inputs_embeds=None,
201
+ cache_position=None,
202
+ position_ids=None,
203
+ use_cache=True,
204
+ **kwargs,
205
+ ):
206
+ point_patches = kwargs.get("point_patches")
207
+ point_patch_indices = kwargs.get("point_patch_indices")
208
+
209
+ if past_key_values is not None:
210
+ if inputs_embeds is not None:
211
+ input_ids = input_ids[:, -cache_position.shape[0]:]
212
+ if point_patch_indices is not None:
213
+ point_patch_indices = point_patch_indices[:, -cache_position.shape[0]:]
214
+ elif input_ids.shape[1] != cache_position.shape[0]:
215
+ input_ids = input_ids[:, cache_position]
216
+ if point_patch_indices is not None:
217
+ batch_size, orig_len = point_patch_indices.shape
218
+ max_pos = cache_position.max().item()
219
+ if max_pos >= orig_len:
220
+ needed_len = max_pos + 1
221
+ seq_diff = needed_len - orig_len
222
+ padding = torch.full((batch_size, seq_diff), -1,
223
+ dtype=point_patch_indices.dtype,
224
+ device=point_patch_indices.device)
225
+ point_patch_indices = torch.cat([point_patch_indices, padding], dim=1)
226
+
227
+ # 现在安全索引
228
+ point_patch_indices = point_patch_indices[:, cache_position]
229
+
230
+ if attention_mask is not None and position_ids is None:
231
+ position_ids = attention_mask.long().cumsum(-1) - 1
232
+ position_ids.masked_fill_(attention_mask == 0, 1)
233
+ if past_key_values:
234
+ position_ids = position_ids[:, -input_ids.shape[1]:]
235
+
236
+ if inputs_embeds is not None and cache_position[0] == 0:
237
+ model_inputs = {"inputs_embeds": inputs_embeds}
238
+ else:
239
+ model_inputs = {"input_ids": input_ids.contiguous()}
240
+
241
+ model_inputs.update(
242
+ {
243
+ "position_ids": position_ids,
244
+ "cache_position": cache_position,
245
+ "past_key_values": past_key_values,
246
+ "use_cache": use_cache,
247
+ "attention_mask": attention_mask,
248
+ }
249
+ )
250
+ if cache_position is None or (cache_position is not None and cache_position[0] == 0):
251
+ if point_patches is not None:
252
+ model_inputs["point_patches"] = point_patches
253
+ if point_patch_indices is not None:
254
+ model_inputs["point_patch_indices"] = point_patch_indices
255
+
256
+ return model_inputs
special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|object_ref_start|>",
6
+ "<|object_ref_end|>",
7
+ "<|box_start|>",
8
+ "<|box_end|>",
9
+ "<|quad_start|>",
10
+ "<|quad_end|>",
11
+ "<|vision_start|>",
12
+ "<|vision_end|>",
13
+ "<|vision_pad|>",
14
+ "<|image_pad|>",
15
+ "<|video_pad|>"
16
+ ],
17
+ "eos_token": {
18
+ "content": "<|im_end|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:930d2050e80a439d4958614ea0a86474a8b0708619e94c90603ef890339440e9
3
+ size 11422839
tokenizer_config.json ADDED
@@ -0,0 +1,248 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "added_tokens_decoder": {
5
+ "151643": {
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "151644": {
14
+ "content": "<|im_start|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "151645": {
22
+ "content": "<|im_end|>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "151646": {
30
+ "content": "<|object_ref_start|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "151647": {
38
+ "content": "<|object_ref_end|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ },
45
+ "151648": {
46
+ "content": "<|box_start|>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false,
51
+ "special": true
52
+ },
53
+ "151649": {
54
+ "content": "<|box_end|>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false,
59
+ "special": true
60
+ },
61
+ "151650": {
62
+ "content": "<|quad_start|>",
63
+ "lstrip": false,
64
+ "normalized": false,
65
+ "rstrip": false,
66
+ "single_word": false,
67
+ "special": true
68
+ },
69
+ "151651": {
70
+ "content": "<|quad_end|>",
71
+ "lstrip": false,
72
+ "normalized": false,
73
+ "rstrip": false,
74
+ "single_word": false,
75
+ "special": true
76
+ },
77
+ "151652": {
78
+ "content": "<|vision_start|>",
79
+ "lstrip": false,
80
+ "normalized": false,
81
+ "rstrip": false,
82
+ "single_word": false,
83
+ "special": true
84
+ },
85
+ "151653": {
86
+ "content": "<|vision_end|>",
87
+ "lstrip": false,
88
+ "normalized": false,
89
+ "rstrip": false,
90
+ "single_word": false,
91
+ "special": true
92
+ },
93
+ "151654": {
94
+ "content": "<|vision_pad|>",
95
+ "lstrip": false,
96
+ "normalized": false,
97
+ "rstrip": false,
98
+ "single_word": false,
99
+ "special": true
100
+ },
101
+ "151655": {
102
+ "content": "<|image_pad|>",
103
+ "lstrip": false,
104
+ "normalized": false,
105
+ "rstrip": false,
106
+ "single_word": false,
107
+ "special": true
108
+ },
109
+ "151656": {
110
+ "content": "<|video_pad|>",
111
+ "lstrip": false,
112
+ "normalized": false,
113
+ "rstrip": false,
114
+ "single_word": false,
115
+ "special": true
116
+ },
117
+ "151657": {
118
+ "content": "<tool_call>",
119
+ "lstrip": false,
120
+ "normalized": false,
121
+ "rstrip": false,
122
+ "single_word": false,
123
+ "special": false
124
+ },
125
+ "151658": {
126
+ "content": "</tool_call>",
127
+ "lstrip": false,
128
+ "normalized": false,
129
+ "rstrip": false,
130
+ "single_word": false,
131
+ "special": false
132
+ },
133
+ "151659": {
134
+ "content": "<|fim_prefix|>",
135
+ "lstrip": false,
136
+ "normalized": false,
137
+ "rstrip": false,
138
+ "single_word": false,
139
+ "special": false
140
+ },
141
+ "151660": {
142
+ "content": "<|fim_middle|>",
143
+ "lstrip": false,
144
+ "normalized": false,
145
+ "rstrip": false,
146
+ "single_word": false,
147
+ "special": false
148
+ },
149
+ "151661": {
150
+ "content": "<|fim_suffix|>",
151
+ "lstrip": false,
152
+ "normalized": false,
153
+ "rstrip": false,
154
+ "single_word": false,
155
+ "special": false
156
+ },
157
+ "151662": {
158
+ "content": "<|fim_pad|>",
159
+ "lstrip": false,
160
+ "normalized": false,
161
+ "rstrip": false,
162
+ "single_word": false,
163
+ "special": false
164
+ },
165
+ "151663": {
166
+ "content": "<|repo_name|>",
167
+ "lstrip": false,
168
+ "normalized": false,
169
+ "rstrip": false,
170
+ "single_word": false,
171
+ "special": false
172
+ },
173
+ "151664": {
174
+ "content": "<|file_sep|>",
175
+ "lstrip": false,
176
+ "normalized": false,
177
+ "rstrip": false,
178
+ "single_word": false,
179
+ "special": false
180
+ },
181
+ "151665": {
182
+ "content": "<pointcloud>",
183
+ "lstrip": false,
184
+ "normalized": true,
185
+ "rstrip": false,
186
+ "single_word": false,
187
+ "special": false
188
+ },
189
+ "151666": {
190
+ "content": "</pointcloud>",
191
+ "lstrip": false,
192
+ "normalized": true,
193
+ "rstrip": false,
194
+ "single_word": false,
195
+ "special": false
196
+ },
197
+ "151667": {
198
+ "content": "<point_patch>",
199
+ "lstrip": false,
200
+ "normalized": true,
201
+ "rstrip": false,
202
+ "single_word": false,
203
+ "special": false
204
+ },
205
+ "151668": {
206
+ "content": "<row_sep>",
207
+ "lstrip": false,
208
+ "normalized": true,
209
+ "rstrip": false,
210
+ "single_word": false,
211
+ "special": false
212
+ },
213
+ "151669": {
214
+ "content": "<layer_sep>",
215
+ "lstrip": false,
216
+ "normalized": true,
217
+ "rstrip": false,
218
+ "single_word": false,
219
+ "special": false
220
+ }
221
+ },
222
+ "additional_special_tokens": [
223
+ "<|im_start|>",
224
+ "<|im_end|>",
225
+ "<|object_ref_start|>",
226
+ "<|object_ref_end|>",
227
+ "<|box_start|>",
228
+ "<|box_end|>",
229
+ "<|quad_start|>",
230
+ "<|quad_end|>",
231
+ "<|vision_start|>",
232
+ "<|vision_end|>",
233
+ "<|vision_pad|>",
234
+ "<|image_pad|>",
235
+ "<|video_pad|>"
236
+ ],
237
+ "bos_token": null,
238
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\n\\n# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0]['role'] == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n {%- else %}\n {{- '<|im_start|>system\\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
239
+ "clean_up_tokenization_spaces": false,
240
+ "eos_token": "<|im_end|>",
241
+ "errors": "replace",
242
+ "extra_special_tokens": {},
243
+ "model_max_length": 131072,
244
+ "pad_token": "<|endoftext|>",
245
+ "split_special_tokens": false,
246
+ "tokenizer_class": "Qwen2Tokenizer",
247
+ "unk_token": null
248
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff