yi-01-ai
commited on
Commit
•
ba1bbc8
1
Parent(s):
3aead14
Auto Sync from git://github.com/01-ai/Yi.git/commit/ed84c741ee48d104263fda3de1a71c15b0f12d53
Browse files
README.md
CHANGED
@@ -122,7 +122,7 @@ pipeline_tag: text-generation
|
|
122 |
- For Chinese language capability, the Yi series models landed in 2nd place (following GPT-4), surpassing other LLMs (such as Baidu ERNIE, Qwen, and Baichuan) on the [SuperCLUE](https://www.superclueai.com/) in Oct 2023.
|
123 |
|
124 |
- 🙏 (Credits to LLaMA) Thanks to the Transformer and LLaMA open-source communities, as they reducing the efforts required to build from scratch and enabling the utilization of the same tools within the AI ecosystem.
|
125 |
-
<details style="display: inline;"><summary> If you're interested in Yi's adoption of LLaMA architecture and license usage policy, see <span style="color: green;">Yi's relation with LLaMA
|
126 |
> 💡 TL;DR
|
127 |
>
|
128 |
> The Yi series models adopt the same model architecture as LLaMA but are **NOT** derivatives of LLaMA.
|
@@ -193,7 +193,7 @@ Yi-6B-200K | • [🤗 Hugging Face](https://huggingface.co/01-ai/Yi-6B-200K)
|
|
193 |
|
194 |
- For chat models:
|
195 |
|
196 |
-
<details style="display: inline;"><summary>For chat model limitations, see ⬇️</summary>
|
197 |
<ul>
|
198 |
<br>The released chat model has undergone exclusive training using Supervised Fine-Tuning (SFT). Compared to other standard chat models, our model produces more diverse responses, making it suitable for various downstream tasks, such as creative scenarios. Furthermore, this diversity is expected to enhance the likelihood of generating higher quality responses, which will be advantageous for subsequent Reinforcement Learning (RL) training.
|
199 |
|
@@ -414,7 +414,7 @@ Then you can see an output similar to the one below. 🥳
|
|
414 |
|
415 |
<details>
|
416 |
|
417 |
-
<summary>Output ⬇️ </summary>
|
418 |
|
419 |
<br>
|
420 |
|
@@ -426,7 +426,7 @@ Then you can see an output similar to the one below. 🥳
|
|
426 |
|
427 |
### Quick start - Docker
|
428 |
<details>
|
429 |
-
<summary> Run Yi-34B-chat locally with Docker: a step-by-step guide ⬇️</summary>
|
430 |
<br>This tutorial guides you through every step of running <strong>Yi-34B-Chat on an A800 GPU</strong> locally and then performing inference.
|
431 |
<h4>Step 0: Prerequisites</h4>
|
432 |
<p>Make sure you've installed <a href="https://docs.docker.com/engine/install/?open_in_browser=true">Docker</a> and <a href="https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html">nvidia-container-toolkit</a>.</p>
|
@@ -453,7 +453,7 @@ ghcr.io/01-ai/yi:latest
|
|
453 |
|
454 |
### Quick start - llama.cpp
|
455 |
<details>
|
456 |
-
<summary> Run Yi-chat-6B-2bits locally with llama.cpp: a step-by-step guide ⬇️</summary>
|
457 |
<br>This tutorial guides you through every step of running a quantized model (<a href="https://huggingface.co/XeIaso/yi-chat-6B-GGUF/tree/main">Yi-chat-6B-2bits</a>) locally and then performing inference.</p>
|
458 |
|
459 |
- [Step 0: Prerequisites](#step-0-prerequisites)
|
@@ -602,7 +602,7 @@ python demo/web_demo.py -c <your-model-path>
|
|
602 |
|
603 |
You can access the web UI by entering the address provided in the console into your browser.
|
604 |
|
605 |
-
![Quick start - web demo](
|
606 |
|
607 |
### Finetuning
|
608 |
|
@@ -615,7 +615,7 @@ Once finished, you can compare the finetuned model and the base model with the f
|
|
615 |
```bash
|
616 |
bash finetune/scripts/run_eval.sh
|
617 |
```
|
618 |
-
<details style="display: inline;"><summary>For advanced usage (like fine-tuning based on your custom data), see
|
619 |
|
620 |
### Finetune code for Yi 6B and 34B
|
621 |
|
@@ -747,7 +747,7 @@ python quantization/gptq/eval_quantized_model.py \
|
|
747 |
--trust_remote_code
|
748 |
```
|
749 |
|
750 |
-
<details style="display: inline;"><summary>For a more detailed explanation, see ⬇️</summary> <ul>
|
751 |
|
752 |
#### GPT-Q quantization
|
753 |
|
@@ -798,7 +798,7 @@ python quantization/awq/eval_quantized_model.py \
|
|
798 |
--model /quantized_model \
|
799 |
--trust_remote_code
|
800 |
```
|
801 |
-
<details style="display: inline;"><summary>For detailed explanations, see ⬇️</summary> <ul>
|
802 |
|
803 |
#### AWQ quantization
|
804 |
|
@@ -883,7 +883,7 @@ Below are detailed minimum VRAM requirements under different batch use cases.
|
|
883 |
### Learning hub
|
884 |
|
885 |
<details>
|
886 |
-
<summary> If you want to learn Yi, you can find a wealth of helpful educational resources here ⬇️</summary>
|
887 |
<br>
|
888 |
|
889 |
Welcome to the Yi learning hub!
|
@@ -1010,10 +1010,10 @@ If you're seeking to explore the diverse capabilities within Yi's thriving famil
|
|
1010 |
|
1011 |
Yi-34B-Chat model demonstrates exceptional performance, ranking first among all existing open-source models in the benchmarks including MMLU, CMMLU, BBH, GSM8k, and more.
|
1012 |
|
1013 |
-
![Chat model performance](
|
1014 |
|
1015 |
<details>
|
1016 |
-
<summary> Evaluation methods and challenges ⬇️ </summary>
|
1017 |
|
1018 |
- **Evaluation methods**: we evaluated various benchmarks using both zero-shot and few-shot methods, except for TruthfulQA.
|
1019 |
- **Zero-shot vs. few-shot**: in chat models, the zero-shot approach is more commonly employed.
|
@@ -1027,18 +1027,18 @@ Yi-34B-Chat model demonstrates exceptional performance, ranking first among all
|
|
1027 |
|
1028 |
The Yi-34B and Yi-34B-200K models stand out as the top performers among open-source models, especially excelling in MMLU, CMML, common-sense reasoning, reading comprehension, and more.
|
1029 |
|
1030 |
-
![Base model performance](
|
1031 |
|
1032 |
<details>
|
1033 |
-
<summary> Evaluation methods ⬇️</summary>
|
1034 |
-
|
1035 |
-
- **Disparity in
|
1036 |
-
- **Investigation
|
1037 |
-
- **Uniform
|
1038 |
-
- **Efforts to
|
1039 |
-
- **Extensive
|
1040 |
-
- **Special
|
1041 |
-
- **Falcon-180B
|
1042 |
</details>
|
1043 |
|
1044 |
# 🟢 Who can use Yi?
|
|
|
122 |
- For Chinese language capability, the Yi series models landed in 2nd place (following GPT-4), surpassing other LLMs (such as Baidu ERNIE, Qwen, and Baichuan) on the [SuperCLUE](https://www.superclueai.com/) in Oct 2023.
|
123 |
|
124 |
- 🙏 (Credits to LLaMA) Thanks to the Transformer and LLaMA open-source communities, as they reducing the efforts required to build from scratch and enabling the utilization of the same tools within the AI ecosystem.
|
125 |
+
<details style="display: inline;"><summary> If you're interested in Yi's adoption of LLaMA architecture and license usage policy, see <span style="color: green;">Yi's relation with LLaMA.</span> ⬇️</summary> <ul> <br>
|
126 |
> 💡 TL;DR
|
127 |
>
|
128 |
> The Yi series models adopt the same model architecture as LLaMA but are **NOT** derivatives of LLaMA.
|
|
|
193 |
|
194 |
- For chat models:
|
195 |
|
196 |
+
<details style="display: inline;"><summary>For chat model limitations, see the explanations below. ⬇️</summary>
|
197 |
<ul>
|
198 |
<br>The released chat model has undergone exclusive training using Supervised Fine-Tuning (SFT). Compared to other standard chat models, our model produces more diverse responses, making it suitable for various downstream tasks, such as creative scenarios. Furthermore, this diversity is expected to enhance the likelihood of generating higher quality responses, which will be advantageous for subsequent Reinforcement Learning (RL) training.
|
199 |
|
|
|
414 |
|
415 |
<details>
|
416 |
|
417 |
+
<summary>Output. ⬇️ </summary>
|
418 |
|
419 |
<br>
|
420 |
|
|
|
426 |
|
427 |
### Quick start - Docker
|
428 |
<details>
|
429 |
+
<summary> Run Yi-34B-chat locally with Docker: a step-by-step guide. ⬇️</summary>
|
430 |
<br>This tutorial guides you through every step of running <strong>Yi-34B-Chat on an A800 GPU</strong> locally and then performing inference.
|
431 |
<h4>Step 0: Prerequisites</h4>
|
432 |
<p>Make sure you've installed <a href="https://docs.docker.com/engine/install/?open_in_browser=true">Docker</a> and <a href="https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html">nvidia-container-toolkit</a>.</p>
|
|
|
453 |
|
454 |
### Quick start - llama.cpp
|
455 |
<details>
|
456 |
+
<summary> Run Yi-chat-6B-2bits locally with llama.cpp: a step-by-step guide. ⬇️</summary>
|
457 |
<br>This tutorial guides you through every step of running a quantized model (<a href="https://huggingface.co/XeIaso/yi-chat-6B-GGUF/tree/main">Yi-chat-6B-2bits</a>) locally and then performing inference.</p>
|
458 |
|
459 |
- [Step 0: Prerequisites](#step-0-prerequisites)
|
|
|
602 |
|
603 |
You can access the web UI by entering the address provided in the console into your browser.
|
604 |
|
605 |
+
![Quick start - web demo](https://github.com/01-ai/Yi/blob/main/assets/img/yi_34b_chat_web_demo.gif)
|
606 |
|
607 |
### Finetuning
|
608 |
|
|
|
615 |
```bash
|
616 |
bash finetune/scripts/run_eval.sh
|
617 |
```
|
618 |
+
<details style="display: inline;"><summary>For advanced usage (like fine-tuning based on your custom data), see the explanations below. ⬇️ </summary> <ul>
|
619 |
|
620 |
### Finetune code for Yi 6B and 34B
|
621 |
|
|
|
747 |
--trust_remote_code
|
748 |
```
|
749 |
|
750 |
+
<details style="display: inline;"><summary>For a more detailed explanation, see the explanations below. ⬇️</summary> <ul>
|
751 |
|
752 |
#### GPT-Q quantization
|
753 |
|
|
|
798 |
--model /quantized_model \
|
799 |
--trust_remote_code
|
800 |
```
|
801 |
+
<details style="display: inline;"><summary>For detailed explanations, see the explanations below. ⬇️</summary> <ul>
|
802 |
|
803 |
#### AWQ quantization
|
804 |
|
|
|
883 |
### Learning hub
|
884 |
|
885 |
<details>
|
886 |
+
<summary> If you want to learn Yi, you can find a wealth of helpful educational resources here. ⬇️</summary>
|
887 |
<br>
|
888 |
|
889 |
Welcome to the Yi learning hub!
|
|
|
1010 |
|
1011 |
Yi-34B-Chat model demonstrates exceptional performance, ranking first among all existing open-source models in the benchmarks including MMLU, CMMLU, BBH, GSM8k, and more.
|
1012 |
|
1013 |
+
![Chat model performance](https://github.com/01-ai/Yi/blob/main/assets/img/benchmark_base.png)
|
1014 |
|
1015 |
<details>
|
1016 |
+
<summary> Evaluation methods and challenges. ⬇️ </summary>
|
1017 |
|
1018 |
- **Evaluation methods**: we evaluated various benchmarks using both zero-shot and few-shot methods, except for TruthfulQA.
|
1019 |
- **Zero-shot vs. few-shot**: in chat models, the zero-shot approach is more commonly employed.
|
|
|
1027 |
|
1028 |
The Yi-34B and Yi-34B-200K models stand out as the top performers among open-source models, especially excelling in MMLU, CMML, common-sense reasoning, reading comprehension, and more.
|
1029 |
|
1030 |
+
![Base model performance](https://github.com/01-ai/Yi/blob/main/assets/img/benchmark_base.png)
|
1031 |
|
1032 |
<details>
|
1033 |
+
<summary> Evaluation methods. ⬇️</summary>
|
1034 |
+
|
1035 |
+
- **Disparity in results**: while benchmarking open-source models, a disparity has been noted between results from our pipeline and those reported by public sources like OpenCompass.
|
1036 |
+
- **Investigation findings**: a deeper investigation reveals that variations in prompts, post-processing strategies, and sampling techniques across models may lead to significant outcome differences.
|
1037 |
+
- **Uniform benchmarking process**: our methodology aligns with the original benchmarks—consistent prompts and post-processing strategies are used, and greedy decoding is applied during evaluations without any post-processing for the generated content.
|
1038 |
+
- **Efforts to retrieve unreported scores**: for scores that were not reported by the original authors (including scores reported with different settings), we try to get results with our pipeline.
|
1039 |
+
- **Extensive model evaluation**: to evaluate the model’s capability extensively, we adopted the methodology outlined in Llama2. Specifically, we included PIQA, SIQA, HellaSwag, WinoGrande, ARC, OBQA, and CSQA to assess common sense reasoning. SquAD, QuAC, and BoolQ were incorporated to evaluate reading comprehension.
|
1040 |
+
- **Special configurations**: CSQA was exclusively tested using a 7-shot setup, while all other tests were conducted with a 0-shot configuration. Additionally, we introduced GSM8K (8-shot@1), MATH (4-shot@1), HumanEval (0-shot@1), and MBPP (3-shot@1) under the category "Math & Code".
|
1041 |
+
- **Falcon-180B caveat**: Falcon-180B was not tested on QuAC and OBQA due to technical constraints. Its performance score is an average from other tasks, and considering the generally lower scores of these two tasks, Falcon-180B's capabilities are likely not underestimated.
|
1042 |
</details>
|
1043 |
|
1044 |
# 🟢 Who can use Yi?
|