feihu.hf
commited on
Commit
·
098145f
1
Parent(s):
83c0546
update README & config.json
Browse files- README.md +23 -1
- config.json +1 -1
README.md
CHANGED
|
@@ -34,7 +34,8 @@ Qwen2.5-Coder is the latest series of Code-Specific Qwen large language models (
|
|
| 34 |
- Number of Paramaters (Non-Embedding): 1.31B
|
| 35 |
- Number of Layers: 28
|
| 36 |
- Number of Attention Heads (GQA): 12 for Q and 2 for KV
|
| 37 |
-
- Context Length: Full
|
|
|
|
| 38 |
|
| 39 |
For more details, please refer to our [blog](https://qwenlm.github.io/blog/qwen2.5-coder/), [GitHub](https://github.com/QwenLM/Qwen2.5-Coder), and [Documentation](https://qwen.readthedocs.io/en/latest/).
|
| 40 |
|
|
@@ -86,6 +87,27 @@ generated_ids = [
|
|
| 86 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 87 |
```
|
| 88 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
|
| 90 |
## Evaluation & Performance
|
| 91 |
|
|
|
|
| 34 |
- Number of Paramaters (Non-Embedding): 1.31B
|
| 35 |
- Number of Layers: 28
|
| 36 |
- Number of Attention Heads (GQA): 12 for Q and 2 for KV
|
| 37 |
+
- Context Length: Full 131,072 tokens
|
| 38 |
+
- Please refer to [this section](#processing-long-texts) for detailed instructions on how to deploy Qwen2.5 for handling long texts.
|
| 39 |
|
| 40 |
For more details, please refer to our [blog](https://qwenlm.github.io/blog/qwen2.5-coder/), [GitHub](https://github.com/QwenLM/Qwen2.5-Coder), and [Documentation](https://qwen.readthedocs.io/en/latest/).
|
| 41 |
|
|
|
|
| 87 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
| 88 |
```
|
| 89 |
|
| 90 |
+
### Processing Long Texts
|
| 91 |
+
|
| 92 |
+
The current `config.json` is set for context length up to 32,768 tokens.
|
| 93 |
+
To handle extensive inputs exceeding 32,768 tokens, we utilize [YaRN](https://arxiv.org/abs/2309.00071), a technique for enhancing model length extrapolation, ensuring optimal performance on lengthy texts.
|
| 94 |
+
|
| 95 |
+
For supported frameworks, you could add the following to `config.json` to enable YaRN:
|
| 96 |
+
```json
|
| 97 |
+
{
|
| 98 |
+
...,
|
| 99 |
+
"rope_scaling": {
|
| 100 |
+
"factor": 4.0,
|
| 101 |
+
"original_max_position_embeddings": 32768,
|
| 102 |
+
"type": "yarn"
|
| 103 |
+
}
|
| 104 |
+
}
|
| 105 |
+
```
|
| 106 |
+
|
| 107 |
+
For deployment, we recommend using vLLM.
|
| 108 |
+
Please refer to our [Documentation](https://qwen.readthedocs.io/en/latest/deployment/vllm.html) for usage if you are not familar with vLLM.
|
| 109 |
+
Presently, vLLM only supports static YARN, which means the scaling factor remains constant regardless of input length, **potentially impacting performance on shorter texts**.
|
| 110 |
+
We advise adding the `rope_scaling` configuration only when processing long contexts is required.
|
| 111 |
|
| 112 |
## Evaluation & Performance
|
| 113 |
|
config.json
CHANGED
|
@@ -17,7 +17,7 @@
|
|
| 17 |
"num_key_value_heads": 2,
|
| 18 |
"rms_norm_eps": 1e-06,
|
| 19 |
"rope_theta": 1000000.0,
|
| 20 |
-
"sliding_window":
|
| 21 |
"tie_word_embeddings": true,
|
| 22 |
"torch_dtype": "bfloat16",
|
| 23 |
"transformers_version": "4.44.0",
|
|
|
|
| 17 |
"num_key_value_heads": 2,
|
| 18 |
"rms_norm_eps": 1e-06,
|
| 19 |
"rope_theta": 1000000.0,
|
| 20 |
+
"sliding_window": 131072,
|
| 21 |
"tie_word_embeddings": true,
|
| 22 |
"torch_dtype": "bfloat16",
|
| 23 |
"transformers_version": "4.44.0",
|