GitHub | 🤖 ModelScope | 📑 ArXiv
Hare-1.1B-base is a pre-trained model developed by the LiteAI Team from China Telecom Guizhou Branch. We use a mix of high-quality open-source data and strategy-generated data as pre-train data. The model is only 1.1B in size and has performed well on the Open LLM Leaderboard. - We chose Mistral as the foundational architecture and reused its tokenizer, reducing the number of parameters by adjusting the hyperparameters of its model architecture. Consequently, our model can be directly applied to numerous open-source projects that support Mistral, such as vLLM. - Our model has a parameter count of only 1.1 billion, allowing us to deploy it on consumer-grade GPUs, mobile devices, and other cost-effective platforms. - We have explored efficient training at FP8 precision and have compiled a set of best practices, hoping to contribute as much as we can to LLM training in the open-source community. For best practices, please see our GitHub homepage. - We are currently developing and adapting for Chinese language support. Hare-1.1B-base是由中国电信股份有限公司贵州分公司LiteAI团队开发的预训练模型。我们使用高质量开源和策略生成的合成数据作为预训练数据。该模型大小仅为1.1B，并在Open LLM Leaderboard上表现优异。 - 我们选择Mistral架构作为基础框架，并复用了其分词器，通过调整模型架构的超参来减少参数量。因此，我们的模型可以直接应用于许多支持Mistral的开源项目，如vLLM。 - 我们模型的参数量仅为 11 亿，因此，我们可以将模型部署到消费级显卡、手机端等成本较低的设备上。 - 我们探索了FP8精度下的高效训练，并总结了一份最佳实践，希望能为开源社区LLM训练作出力所能及的贡献。最佳实践请看GitHub主页。 - 我们正在研发与适配中文。 ## Model Details 模型细节 | Model | Training Tokens | Hidden layers | Hidden Size | Attention Heads | Context Length | |:------:|:--------:|:---------:|:-------------:|:-----------------:|:----------------:| |Hare-1.1B-base | ~ 600B |22 | 2048 | 32 | 2048 | ## Model Description 模型说明 - **Developed by:** LiteAI Team - **Institution:** China Telecom Guizhou Branch - **Model size:** 1.1B - **License:** Apache 2.0 - **开发者:** LiteAI Team - **机构:** 中国电信股份有限公司贵州分公司 - **模型大小:** 1.1B - **协议:** Apache 2.0 ## Uses 模型使用 ### Inference 推理 ```python import torch from transformers import AutoTokenizer, AutoModelForCausalLM device = "cuda" if torch.cuda.is_available() else "cpu" model_path = "LiteAI-Team/Hare-1.1B-base" tokenizer = AutoTokenizer.from_pretrained(model_path) model = AutoModelForCausalLM.from_pretrained(model_path) model.to(device) prompt = "Write a poem based on the landscape of Guizhou:" tokens = tokenizer(prompt, add_special_tokens=True, return_tensors='pt').to(device) output = model.generate(**tokens,max_new_tokens=128) output_tokens = output[0].cpu().numpy()[tokens.input_ids.size()[1]:] output_string = tokenizer.decode(output_tokens) print(output_string) >> """The Guizhou landscape is a sight to behold, A place where nature's beauty is unmatched, A land of towering mountains and vast plains, A paradise for those who seek to explore. The mountains rise high above the sky, A sight to beholder, a sight to see, The valleys stretch out as far as the eye can see, A landscape of endless beauty and grace.""" ``` Install with vllm: ```shell pip install vllm ``` ```python from vllm import LLM, SamplingParams from transformers import AutoTokenizer model_path = "LiteAI-Team/Hare-1.1B-base" llm = LLM(model=model_path, trust_remote_code=True, tensor_parallel_size=4) query = "Write a poem based on the landscape of Guizhou:" sampling_params = SamplingParams(temperature=0.8, top_p=0.95, max_tokens=64) outputs = llm.generate(query, sampling_params) print(outputs) ``` ## Edge Deployment Demo 端侧部署 Our model has only 1.1 billion parameters, and after Int4 quantization, it occupies just 0.6GB of space, allowing for easy deployment on mobile devices, The [Hare-1.1B-Chat](https://huggingface.co/LiteAI/Hare-1.1B-Chat) model weights have been open-sourced. - Android：We chose MLC-LLM as the deployment framework and conducted deployment testing of the Chat model on the Redmi K40. - iOS & HarmonyOS：We will conduct deployment testing on the aforementioned devices in the future. 我们的模型参数量仅有1.1B，经Int4量化后，模型仅占用0.6G的空间，可轻松部署在手机端，[Hare-1.1B-Chat](https://huggingface.co/LiteAI/Hare-1.1B-Chat)模型权重已经开源。 - Android：我们选择MLC-LLM作为部署框架，在Redmi K40上进行Chat模型的部署测试。 - iOS & HarmonyOS：我们将在未来对上述设备进行部署测试。 |
 |
](https://www.bilibili.com/video/BV1Ry411b7yx/?vd_source=d4f08e4b18c51571a1b53a20a8d58c10)
## Evaluation Results 评测结果
- Additionally, we conducted explorations and experiments addressing the issue of benchmark data leakage. For a detailed analysis, please refer to our [paper](https://arxiv.org/abs/2406.11410).
- 同时，我们针对benchmark数据泄漏问题做了探索与实验，详细分析请参考我们的[论文](https://arxiv.org/abs/2406.11410)。
| Model(base) | Size | avg | MMLU | ARC-C | TruthfulQA | Winogrande | Hellaswag | GSM8K |
|:-------------------------------------:|:-------:|:-------:|:------:|:-------:|:------------:|:------------:|:-----------:|:-------:|
| phi-1_5 | 1.3B | 47.69 | 43.89| 52.9 | 40.89 | 72.22 | 63.79 |12.43 |
| Qwen-1.5 | 1.8B | 46.55 | 46.71| 37.88 | 39.43 | 60.3 | 61.42 |33.59 |
| stablelm-2 | 1.6B | 45.25 | 38.95| 43.34 | 36.78 | 64.56 | 70.45 |17.44 |
| __Hare__ | 1.1B | 40.17 | 35.74| 38.4 | 42.08 | 59.27 | 57.46 |8.04 |
| H2o-danube | 1.8B | 39.12 | 25.94| 39.42 | 33.86 | 64.48 | 69.58 |1.44 |
| OpenELM | 1.1B | 38.47 | 27.05| 36.69 | 33.86 | 63.22 | 65.71 |1.21 |
| csg-wukong | 1B | 37.78 | 25.33| 37.71 | 42.79 | 56.67 | 58.93 |5.23 |
| TinyLlama-3T | 1.1B | 36.42 | 26.04| 33.87 | 37.32 | 59.51 | 60.31 |1.44 |
## License 协议
- This repository is open-sourced under the Apache-2.0 license.
- The Hare series model weights are currently fully open only for academic research.
- 本仓库遵循Apache-2.0协议开源。
- Hare系列模型权重目前仅对学术研究完全开放。
## Statement 声明
- Hare is a language model trained on a mix of open-source pre-training data and strategy-generated pre-training data. It lacks the ability to make value judgments and cannot understand or express personal opinions. The outputs of the model do not represent the views or positions of the LiteAI development team.
- Therefore, the content generated using Hare may contain biased viewpoints and inaccuracies. Please use it at your discretion.
- Similarly, we will not assume any responsibility for risks and issues arising from users deliberately using Hare to generate harmful content.
- For modifications related to this repository, please contact: zhangly41 At(@) chinatelecom.cn.
- Team contact information: chensq27 At(@) chinatelecom.cn. The LiteAI Team looks forward to collaborating with you.
- Hare是一个基于开源预训练数据和策略合成预训练数据混合训练得到的语言模型，它不具备价值判断能力，无法理解、表达个人观点，模型的输出内容不代表LiteAI开发团队的观点与立场。
- 因此，您使用Hare生成的内容可能存有偏观点和不实情况，请您酌情使用。
- 同样，我们将不承担用户故意使用Hare进行有害内容生成所带来的任何风险与问题。
- 如涉及到本仓库的修改，请联系：zhangly41 At(@) chinatelecom.cn。
- 团队联系方式：chensq27 At(@) chinatelecom.cn，LiteAI团队期待您的合作。
## Citation 工作引用
- If you find Hare helpful for your work, please consider citing our [paper](https://arxiv.org/abs/2406.11410).
- 如果您觉得Hare对您的工作起到了帮助，请考虑引用我们的[论文](https://arxiv.org/abs/2406.11410)。
```
@misc{zhang2024harehumanpriorskey,
title={HARE: HumAn pRiors, a key to small language model Efficiency},
author={Lingyun Zhang and Bin jin and Gaojian Ge and Lunhui Liu and Xuewen Shen and Mingyong Wu and Houqian Zhang and Yongneng Jiang and Shiqi Chen and Shi Pu},
year={2024},
eprint={2406.11410},
archivePrefix={arXiv},
primaryClass={cs.CL}
url={https://arxiv.org/abs/2406.11410},
}
```