infgrad commited on
Commit
f47c8f7
·
verified ·
1 Parent(s): 0361aa1

Upload 14 files

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
README.md CHANGED
@@ -1,3 +1,150 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ tags:
4
+
5
+ - sentence-transformers
6
+ - sentence-similarity
7
+ - mteb
8
+ - retriever
9
+ - text-embeddings-inference
10
+
11
  ---
12
+
13
+ # Jasper-Token-Compression-V2
14
+
15
+ ## Introduction
16
+
17
+ Inspired by Deepseek-OCR, this is the first vector model in the Jasper and Stella series to use **dynamic text token
18
+ compression**. Through the combination of vector distillation and contrastive learning, our model can compress text by
19
+ 3x
20
+ while still achieving excellent performance!
21
+
22
+ ## Features
23
+
24
+ - ⭐⭐⭐ Supports bilingual (Chinese and English)
25
+ - ⭐⭐⭐⭐⭐⭐ Dynamic token compression - tested to achieve excellent results even when compressing text to 0.33x of original
26
+ length
27
+ - ⭐⭐⭐ Combines vector distillation with contrastive learning to further improve performance on retrieval tasks
28
+ - ⭐⭐ 12 million unsupervised data distillation
29
+ - ⭐⭐ 0.6B parameter size
30
+
31
+ ## Technical Details
32
+
33
+ ### Dynamic Text Token Compression
34
+
35
+ My implementation is very simple: After text passes through the `word_embedding` layer, it immediately goes through a
36
+ `Qwen3MLP` (approximately 3 fully connected layers), then I calculate the compressed length, and finally use
37
+ `adaptive_avg_pool1d` to compress tokens to that length.
38
+
39
+ The compression length calculation logic is as follows:
40
+
41
+ ```python
42
+ real_length = 1000 # Actual token count of the text
43
+ length_threshold = 80 # Compress only if exceeding this threshold
44
+ compression_ratio = 0.333
45
+ if real_length <= length_threshold:
46
+ # No compression
47
+ pass
48
+ else:
49
+ target_length = int(length_threshold + (real_length - length_threshold) * compression_ratio)
50
+ ```
51
+
52
+ For implementation details, please refer to the `modeling_qwen3_jasper.py` file in this directory.
53
+
54
+ ### Vector Distillation + Contrastive Learning
55
+
56
+ First, we compute teacher vectors for texts in the contrastive learning training set, then use the following three
57
+ losses during training:
58
+
59
+ 1. Cosine loss: Standard vector distillation loss
60
+ 2. InfoNCE (hard loss): Standard contrastive learning loss function
61
+ 3. KL divergence (soft loss): KL divergence between student score matrix and teacher score matrix. The score matrix is
62
+ the scores between query and all documents(i.e. positive doc, hard negative docs, other in-batch docs).
63
+
64
+ #### Evaluation
65
+
66
+ My prompt strategy and specific content are consistent with the QZhou model. Please refer to their evaluation
67
+ script: https://github.com/Kingsoft-LLM/QZhou-Embedding
68
+
69
+ ### Usage
70
+
71
+ ```py
72
+ import torch
73
+ from sentence_transformers import SentenceTransformer
74
+
75
+ if __name__ == "__main__":
76
+ model_name_or_path = "infgrad/Jasper-Token-Compression-V2"
77
+ model = SentenceTransformer(
78
+ model_name_or_path,
79
+ model_kwargs={
80
+ "torch_dtype": torch.bfloat16,
81
+ "attn_implementation": "sdpa", # We support flash_attention_2; sdpa; eager
82
+ "trust_remote_code": True
83
+ },
84
+ trust_remote_code=True,
85
+ tokenizer_kwargs={"padding_side": "left"},
86
+ device="cpu",
87
+ )
88
+
89
+ queries = [
90
+ "What is photosynthesis?",
91
+ "Who invented the telephone?",
92
+ ]
93
+ documents = [
94
+ "Photosynthesis is the process by which green plants use sunlight, carbon dioxide, and water to produce glucose and oxygen",
95
+ "Alexander Graham Bell is credited with inventing the first practical telephone in 1876, receiving US patent number 174,465 for his device."
96
+ ]
97
+ # The smaller the compression_ratio parameter, the faster the speed, but the quality will correspondingly decrease.
98
+ # Based on our parameter settings during training and test results, we recommend a range between 0.3-0.8.
99
+ query_embeddings = model.encode(queries, prompt_name="query", normalize_embeddings=True, compression_ratio=0.3333)
100
+ document_embeddings = model.encode(documents, normalize_embeddings=True, compression_ratio=0.3333)
101
+
102
+ similarity = model.similarity(query_embeddings, document_embeddings)
103
+ print(similarity)
104
+
105
+ ```
106
+
107
+ ### Limitations and TODO
108
+
109
+ #### Retrieval performance
110
+
111
+ I found that distilled models struggle to approach the retrieval performance of teacher models, which is why I
112
+ specifically used contrastive learning + distillation learning to enhance the student model. However, I found that while
113
+ the enhanced model showed improvement on retrieval test sets, there is still a significant gap compared to mainstream
114
+ models.
115
+ **Therefore, I believe that how to improve the retrieval performance of distilled models is a very necessary and
116
+ valuable
117
+ research direction.**
118
+
119
+ #### More reasonable text token compression modules
120
+
121
+ There is limited research on text token compression currently, and I have only tried the simplest approach. I believe
122
+ more reasonable text compression modules can definitely be found.
123
+
124
+ #### Text length
125
+
126
+ I only distilled texts up to 1024 tokens in length, so there should be performance degradation when text length exceeds
127
+ 1024.
128
+
129
+ ### Citation
130
+
131
+ If you find our work worth citing, please use the following citation.
132
+
133
+ For distillation, please cite the following paper:
134
+ **Jasper and Stella Technical Report:**
135
+
136
+ ```
137
+
138
+ @misc{zhang2025jasperstelladistillationsota,
139
+ title={Jasper and Stella: distillation of SOTA embedding models},
140
+ author={Dun Zhang and Jiacheng Li and Ziyang Zeng and Fulong Wang},
141
+ year={2025},
142
+ eprint={2412.19048},
143
+ archivePrefix={arXiv},
144
+ primaryClass={cs.IR},
145
+ url={https://arxiv.org/abs/2412.19048},
146
+ }
147
+
148
+ ```
149
+
150
+ For text compression, please cite this link directly. We will consider writing a report later.
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "JasperV2Encoder"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "auto_map": {
8
+ "AutoModel": "modeling_qwen3_jasper.JasperV2Encoder"
9
+ },
10
+ "bos_token_id": 151643,
11
+ "dtype": "bfloat16",
12
+ "eos_token_id": 151643,
13
+ "head_dim": 128,
14
+ "hidden_act": "silu",
15
+ "hidden_size": 1024,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "max_position_embeddings": 32768,
19
+ "max_window_layers": 28,
20
+ "model_type": "qwen3",
21
+ "num_attention_heads": 16,
22
+ "num_hidden_layers": 28,
23
+ "num_key_value_heads": 8,
24
+ "rms_norm_eps": 1e-06,
25
+ "rope_scaling": null,
26
+ "rope_theta": 1000000,
27
+ "sliding_window": null,
28
+ "tie_word_embeddings": true,
29
+ "transformers_version": "4.57.1",
30
+ "use_cache": false,
31
+ "use_sliding_window": false,
32
+ "vocab_size": 151669
33
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "prompts": {
3
+ "query": "Instruct: Given a web search query, retrieve relevant passages that answer the query\nQuery: ",
4
+ "document": ""
5
+ },
6
+ "default_prompt_name": null,
7
+ "similarity_fn_name": "cosine"
8
+ }
configuration.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "framework": "pytorch",
3
+ "task": "text-generation",
4
+ "allow_remote": true
5
+ }
custom_st.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from sentence_transformers.models import Transformer as BaseTransformer
3
+
4
+
5
+ class JasperTransformer(BaseTransformer):
6
+ def forward(self, features: dict[str, torch.Tensor], **kwargs) -> dict[str, torch.Tensor]:
7
+ vectors = self.auto_model(**features, **kwargs)
8
+ features.update({"sentence_embedding": vectors})
9
+ return features
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "eos_token_id": 151643,
4
+ "max_new_tokens": 2048,
5
+ "transformers_version": "4.51.3"
6
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4a7b9e62fcbfcfbffe217d4cf45b4957502df1bce4fbc05a8f13c1f05f33158
3
+ size 1214661536
modeling_qwen3_jasper.py ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import random
2
+
3
+ import torch
4
+ import torch.nn as nn
5
+ import torch.nn.functional as F
6
+ from transformers import Qwen3PreTrainedModel, Qwen3Config, Qwen3Model
7
+ from transformers.models.qwen3.modeling_qwen3 import Qwen3MLP
8
+
9
+
10
+ class TokenCompressor(nn.Module):
11
+ """
12
+ 自适应Token压缩模块
13
+ 对于长度超过阈值的序列,使用adaptive_avg_pool1d进行压缩
14
+ 压缩后长度 = 阈值 + 超出部分 * compression_ratio
15
+ """
16
+
17
+ def __init__(self, length_threshold: int = 512, compression_ratio: float = 0.3):
18
+ super().__init__()
19
+ self.length_threshold = length_threshold
20
+ self.compression_ratio = compression_ratio
21
+
22
+ def forward(
23
+ self, token_embeddings: torch.Tensor, attention_mask: torch.Tensor
24
+ ) -> tuple[torch.Tensor, torch.Tensor]:
25
+ """
26
+ 对token embeddings进行自适应压缩
27
+ Args:
28
+ token_embeddings: [batch_size, seq_len, hidden_size]
29
+ attention_mask: [batch_size, seq_len]
30
+ Returns:
31
+ compressed_embeddings: 压缩后的embeddings
32
+ compressed_mask: 压缩后的attention mask
33
+ """
34
+ padding_side = 'right' if (attention_mask[:, -1] == 0).any() else 'left'
35
+
36
+ compressed_embeddings_list = []
37
+ compressed_masks_list = []
38
+ for text_idx in range(token_embeddings.shape[0]):
39
+ # 获取当前样本的有效长度
40
+ real_length = int(attention_mask[text_idx].sum().item())
41
+ if real_length <= self.length_threshold:
42
+ # 根据padding方向提取有效的token embeddings
43
+ if padding_side == 'left':
44
+ # 左填充:有效tokens在右边
45
+ valid_embeddings = token_embeddings[text_idx:text_idx + 1, -real_length:, :]
46
+ else:
47
+ # 右填充:有效tokens在左边
48
+ valid_embeddings = token_embeddings[text_idx:text_idx + 1, :real_length, :]
49
+ compressed_embeddings_list.append(valid_embeddings)
50
+ compressed_masks_list.append([1] * real_length)
51
+ else:
52
+ target_length = int(
53
+ self.length_threshold + (real_length - self.length_threshold) * self.compression_ratio
54
+ )
55
+ # 根据padding方向提取有效的token embeddings
56
+ if padding_side == 'left':
57
+ # 左填充:有效tokens在右边
58
+ valid_embeddings = token_embeddings[text_idx:text_idx + 1, -real_length:, :]
59
+ else:
60
+ # 右填充:有效tokens在左边
61
+ valid_embeddings = token_embeddings[text_idx:text_idx + 1, :real_length, :]
62
+
63
+ # 使用adaptive_avg_pool1d进行压缩
64
+ compressed_embeddings_list.append(
65
+ F.adaptive_avg_pool1d(
66
+ valid_embeddings.transpose(1, 2), target_length
67
+ ).transpose(1, 2)
68
+ )
69
+ # print("valid_embeddings.shape,target_length,compressed_embeddings_list[-1].shape",valid_embeddings.shape,target_length,compressed_embeddings_list[-1].shape)
70
+ compressed_masks_list.append([1] * target_length)
71
+
72
+ # 重新组合为token_embeddings和attention_mask
73
+ new_seq_len = max((len(_mask) for _mask in compressed_masks_list))
74
+ new_attention_mask = torch.tensor(
75
+ [
76
+ _mask + [0] * (new_seq_len - len(_mask))
77
+ if padding_side == "right"
78
+ else
79
+ [0] * (new_seq_len - len(_mask)) + _mask
80
+ for _mask in compressed_masks_list
81
+ ],
82
+ dtype=torch.long,
83
+ device=token_embeddings.device
84
+ )
85
+
86
+ # 生成新的token_embeddings
87
+ batch_size = token_embeddings.shape[0]
88
+ hidden_size = token_embeddings.shape[2]
89
+ new_token_embeddings = torch.zeros(
90
+ batch_size, new_seq_len, hidden_size,
91
+ dtype=token_embeddings.dtype,
92
+ device=token_embeddings.device
93
+ )
94
+
95
+ for idx, compressed_emb in enumerate(compressed_embeddings_list):
96
+ seq_len = compressed_emb.shape[1]
97
+ if padding_side == "right":
98
+ new_token_embeddings[idx, :seq_len, :] = compressed_emb.squeeze(0)
99
+ else:
100
+ # print("new_token_embeddings.shape,compressed_emb.shape",new_token_embeddings.shape,compressed_emb.shape)
101
+ new_token_embeddings[idx, -seq_len:, :] = compressed_emb.squeeze(0)
102
+
103
+ return new_token_embeddings, new_attention_mask
104
+
105
+
106
+
107
+ class JasperV2Encoder(Qwen3PreTrainedModel):
108
+
109
+ def __init__(self, config: Qwen3Config):
110
+ super().__init__(config)
111
+ self.model = Qwen3Model(config)
112
+ self.jasper_mlp = Qwen3MLP(config=config)
113
+ self.linear_1 = nn.Linear(in_features=config.hidden_size, out_features=2048, bias=True)
114
+ self.token_compressor = TokenCompressor(length_threshold=80, compression_ratio=0.5)
115
+ self.post_init()
116
+
117
+ def forward(
118
+ self,
119
+ input_ids: torch.Tensor,
120
+ attention_mask: torch.Tensor,
121
+ *args,
122
+ **kwargs
123
+ ) -> torch.Tensor:
124
+ # token_embeddings.shape batch_size*seq_len*hidden_size
125
+ token_embeddings = self.model.embed_tokens(input_ids)
126
+ token_embeddings = self.jasper_mlp(token_embeddings)
127
+
128
+ self.token_compressor.compression_ratio = kwargs.get(
129
+ "compression_ratio",
130
+ self.token_compressor.compression_ratio
131
+ )
132
+ compressed_token_embeddings, attention_mask = self.token_compressor(token_embeddings, attention_mask)
133
+ compressed_token_embeddings = self.model(
134
+ inputs_embeds=compressed_token_embeddings, attention_mask=attention_mask
135
+ )["last_hidden_state"]
136
+
137
+ # 生成句向量
138
+ input_mask_expanded = (
139
+ attention_mask.unsqueeze(-1).expand(compressed_token_embeddings.size()).to(
140
+ compressed_token_embeddings.dtype)
141
+ )
142
+ sum_embeddings = torch.sum(compressed_token_embeddings * input_mask_expanded, 1)
143
+ sum_mask = input_mask_expanded.sum(1)
144
+ sum_mask = torch.clamp(sum_mask, min=1e-9)
145
+ vector = sum_embeddings / sum_mask
146
+ return self.linear_1(vector)
modules.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "custom_st.JasperTransformer",
7
+ "kwargs": [
8
+ "compression_ratio"
9
+ ]
10
+ },
11
+ {
12
+ "idx": 1,
13
+ "name": "1",
14
+ "path": "1_Normalize",
15
+ "type": "sentence_transformers.models.Normalize"
16
+ }
17
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 32768,
3
+ "do_lower_case": false,
4
+ "tokenizer_args": {
5
+ "padding_side": "left"
6
+ }
7
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:def76fb086971c7867b829c23a26261e38d9d74e02139253b38aeb9df8b4b50a
3
+ size 11423705
tokenizer_config.json ADDED
@@ -0,0 +1,240 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "<tool_response>",
183
+ "lstrip": false,
184
+ "normalized": false,
185
+ "rstrip": false,
186
+ "single_word": false,
187
+ "special": false
188
+ },
189
+ "151666": {
190
+ "content": "</tool_response>",
191
+ "lstrip": false,
192
+ "normalized": false,
193
+ "rstrip": false,
194
+ "single_word": false,
195
+ "special": false
196
+ },
197
+ "151667": {
198
+ "content": "<think>",
199
+ "lstrip": false,
200
+ "normalized": false,
201
+ "rstrip": false,
202
+ "single_word": false,
203
+ "special": false
204
+ },
205
+ "151668": {
206
+ "content": "</think>",
207
+ "lstrip": false,
208
+ "normalized": false,
209
+ "rstrip": false,
210
+ "single_word": false,
211
+ "special": false
212
+ }
213
+ },
214
+ "additional_special_tokens": [
215
+ "<|im_start|>",
216
+ "<|im_end|>",
217
+ "<|object_ref_start|>",
218
+ "<|object_ref_end|>",
219
+ "<|box_start|>",
220
+ "<|box_end|>",
221
+ "<|quad_start|>",
222
+ "<|quad_end|>",
223
+ "<|vision_start|>",
224
+ "<|vision_end|>",
225
+ "<|vision_pad|>",
226
+ "<|image_pad|>",
227
+ "<|video_pad|>"
228
+ ],
229
+ "bos_token": null,
230
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0].role == 'system' %}\n {{- messages[0].content + '\\n\\n' }}\n {%- endif %}\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 {%- endif %}\n{%- endif %}\n{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}\n{%- for message in messages[::-1] %}\n {%- set index = (messages|length - 1) - loop.index0 %}\n {%- if ns.multi_step_tool and message.role == \"user\" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}\n {%- set ns.multi_step_tool = false %}\n {%- set ns.last_query_index = index %}\n {%- endif %}\n{%- endfor %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {%- set content = message.content %}\n {%- set reasoning_content = '' %}\n {%- if message.reasoning_content is defined and message.reasoning_content is not none %}\n {%- set reasoning_content = message.reasoning_content %}\n {%- else %}\n {%- if '</think>' in message.content %}\n {%- set content = message.content.split('</think>')[-1].lstrip('\\n') %}\n {%- set reasoning_content = message.content.split('</think>')[0].rstrip('\\n').split('<think>')[-1].lstrip('\\n') %}\n {%- endif %}\n {%- endif %}\n {%- if loop.index0 > ns.last_query_index %}\n {%- if loop.last or (not loop.last and reasoning_content) %}\n {{- '<|im_start|>' + message.role + '\\n<think>\\n' + reasoning_content.strip('\\n') + '\\n</think>\\n\\n' + content.lstrip('\\n') }}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- if message.tool_calls %}\n {%- for tool_call in message.tool_calls %}\n {%- if (loop.first and content) or (not loop.first) %}\n {{- '\\n' }}\n {%- endif %}\n {%- if tool_call.function %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {%- if tool_call.arguments is string %}\n {{- tool_call.arguments }}\n {%- else %}\n {{- tool_call.arguments | tojson }}\n {%- endif %}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if loop.first 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 {%- if enable_thinking is defined and enable_thinking is false %}\n {{- '<think>\\n\\n</think>\\n\\n' }}\n {%- endif %}\n{%- endif %}",
231
+ "clean_up_tokenization_spaces": false,
232
+ "eos_token": "<|im_end|>",
233
+ "errors": "replace",
234
+ "extra_special_tokens": {},
235
+ "model_max_length": 131072,
236
+ "pad_token": "<|endoftext|>",
237
+ "split_special_tokens": false,
238
+ "tokenizer_class": "Qwen2Tokenizer",
239
+ "unk_token": null
240
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff