Add files via upload vllm 0.8.4 add zhinao model support
Add files via upload
vllm 0.8.4 add zhinao model support
中文 | English
欢迎访问360智脑官网 https://ai.360.com 体验更多更强大的功能。
🎉🎉🎉我们开源了360智脑大模型的系列工作,本次开源了以下模型:
360智脑大模型特点如下:
本次发布版本和下载链接见下表: | Size | Model | BF16 | Int4| |:-:|-|:-:|:-:| | 7B | 360Zhinao-7B-Base | 🤖 🤗 | | | 7B | 360Zhinao-7B-Chat-4K | 🤖 🤗 | 🤖 🤗 | | 7B | 360Zhinao-7B-Chat-32K | 🤖 🤗 | 🤖 🤗 | | 7B | 360Zhinao-7B-Chat-360K | 🤖 🤗 | 🤖 🤗 | | 325M | 360Zhinao-search | 🤗 | | | 1.8B | 360Zhinao-1.8B-Reranking | 🤗 | |
我们在OpenCompass的主流评测数据集上验证了我们的模型性能,包括C-Eval、AGIEval、MMLU、CMMLU、HellaSwag、MATH、GSM8K、HumanEval、MBPP、BBH、LAMBADA,考察的能力包括自然语言理解、知识、数学计算和推理、代码生成、逻辑推理等。
以上结果,在官方Opencompass上可查询或可复现。
4K和32K的Chat模型使用相同的4K SFT数据训练。
我们采用了两阶段的方式训练长文本模型.
第一阶段:我们增大RoPE base,将上下文长度扩展至32K训练:
第二阶段:我们将上下文长度扩展至360K进行训练,使用数据如下:
我们在多种长度和多种任务的评测Benchmark上验证不同版本模型的性能。
我们使用LongBench验证长文本效果。LongBench是第一个多任务、中英双语、针对大语言模型长文本理解能力的评测基准。LongBench由六大类、二十一个不同的任务组成,我们选择其中与中文长文本应用最密切相关的中文单文档问答、多文档问答、摘要、Few-shot等任务进行评测。
| Model | Avg | 单文档QA | 多文档QA | 摘要 | Few-shot学习 | 代码补全 | | :------------------------ |:---------:|:--------:|:---------:|:---------:|:------------:|:---------:| | GPT-3.5-Turbo-16k | 37.84 | 61.2 | 28.7 | 16 | 29.2 | 54.1 | | ChatGLM2-6B-32k | 37.16 | 51.6 | 37.6 | 16.2 | 27.7 | 52.7 | | ChatGLM3-6B-32k | 44.62 | **62.3** | 44.8 | 17.8 | 42 | 56.2 | | InternLM2-Chat-7B | 42.20 | 56.65 | 29.15 | **17.99** | 43.5 | **63.72** | | Qwen1.5-Chat-7B | 36.75 | 52.85 | 30.08 | 14.28 | 32 | 54.55 | | Qwen1.5-Chat-14B | 39.80 | 60.39 | 27.99 | 14.77 | 37 | 58.87 | | 360Zhinao-7B-Chat-32K | **45.18** | 57.18 | **48.06** | 15.03 | **44** | 61.64 |
大海捞针测试(NeedleInAHaystack)是将关键信息插入一段长文本的不同位置,再对该关键信息提问,从而测试大模型的长文本能力的一种方法。
360Zhinao-7B-Chat-360K在中英文大海捞针中都能达到98%以上的准确率。
英文”大海捞针”(和NeedleInAHaystack相同)
针:The best thing to do in San Francisco is eat a sandwich and sit in Dolores Park on a sunny day.
提问:What is the best thing to do in San Francisco?
中文“大海捞针”
我们仿照SuperCLUE-200K测评基准构造了中文大海捞针:
海:长篇小说。
针:王莽是一名勤奋的店员,他每天凌晨就起床,赶在第一缕阳光照亮大地之前到达店铺,为即将开始的一天做准备。他清扫店铺,整理货架,为顾客提供方便。他对五金的种类和用途了如指掌,无论顾客需要什么,他总能准确地找到。\n然而,他的老板刘秀却总是对他吹毛求疵。刘秀是个挑剔的人,他总能在王莽的工作中找出一些小错误,然后以此为由扣他的工资。他对王莽的工作要求非常严格,甚至有些过分。即使王莽做得再好,刘秀也总能找出一些小问题,让王莽感到非常沮丧。\n王莽虽然对此感到不满,但他并没有放弃。他知道,只有通过自己的努力,才能获得更好的生活。他坚持每天早起,尽管他知道那天可能会再次被刘秀扣工资。他始终保持微笑,尽管他知道刘秀可能会再次对他挑剔。
提问:王莽在谁的手下工作?
简单的示例来说明如何利用🤖 ModelScope和🤗 Transformers快速使用360Zhinao-7B-Base和360Zhinao-7B-Chat
pip install -r requirements.txt
我们推荐安装flash-attention(当前已支持flash attention 2)来提高你的运行效率以及降低显存占用。(flash-attention只是可选项,不安装也可正常运行该项目)
flash-attn >= 2.3.6
FLASH_ATTENTION_FORCE_BUILD=TRUE pip install flash-attn==2.3.6
此代码演示使用transformers快速使用360Zhinao-7B-Base模型进行推理
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers.generation import GenerationConfig MODEL_NAME_OR_PATH = "qihoo360/360Zhinao-7B-Base" tokenizer = AutoTokenizer.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained( MODEL_NAME_OR_PATH, device_map="auto", trust_remote_code=True) generation_config = GenerationConfig.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) inputs = tokenizer('中国二十四节气\n1. 立春\n2. 雨水\n3. 惊蛰\n4. 春分\n5. 清明\n', return_tensors='pt') inputs = inputs.to(model.device) pred = model.generate(input_ids=inputs["input_ids"], generation_config=generation_config) print("outputs:\n", tokenizer.decode(pred.cpu()[0], skip_special_tokens=True))
此代码演示使用transformers快速使用360Zhinao-7B-Chat-4K模型进行推理
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers.generation import GenerationConfig MODEL_NAME_OR_PATH = "qihoo360/360Zhinao-7B-Chat-4K" tokenizer = AutoTokenizer.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained( MODEL_NAME_OR_PATH, device_map="auto", trust_remote_code=True) generation_config = GenerationConfig.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) messages = [] #round-1 messages.append({"role": "user", "content": "介绍一下刘德华"}) response = model.chat(tokenizer=tokenizer, messages=messages, generation_config=generation_config) messages.append({"role": "assistant", "content": response}) print(messages) #round-2 messages.append({"role": "user", "content": "他有什么代表作?"}) response = model.chat(tokenizer=tokenizer, messages=messages, generation_config=generation_config) messages.append({"role": "assistant", "content": response}) print(messages)
此代码演示使用ModelScope快速使用360Zhinao-7B-Base模型进行推理
from modelscope import AutoModelForCausalLM, AutoTokenizer from modelscope import GenerationConfig MODEL_NAME_OR_PATH = "qihoo360/360Zhinao-7B-Base" tokenizer = AutoTokenizer.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained( MODEL_NAME_OR_PATH, device_map="auto", trust_remote_code=True) generation_config = GenerationConfig.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) inputs = tokenizer('中国二十四节气\n1. 立春\n2. 雨水\n3. 惊蛰\n4. 春分\n5. 清明\n', return_tensors='pt') inputs = inputs.to(model.device) pred = model.generate(input_ids=inputs["input_ids"], generation_config=generation_config) print("outputs:\n", tokenizer.decode(pred.cpu()[0], skip_special_tokens=True))
此代码演示使用ModelScope快速使用360Zhinao-7B-Chat-4K模型进行推理
from modelscope import AutoModelForCausalLM, AutoTokenizer from modelscope import GenerationConfig MODEL_NAME_OR_PATH = "qihoo360/360Zhinao-7B-Chat-4K" tokenizer = AutoTokenizer.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained( MODEL_NAME_OR_PATH, device_map="auto", trust_remote_code=True) generation_config = GenerationConfig.from_pretrained( MODEL_NAME_OR_PATH, trust_remote_code=True) messages = [] #round-1 messages.append({"role": "user", "content": "介绍一下刘德华"}) response = model.chat(tokenizer=tokenizer, messages=messages, generation_config=generation_config) messages.append({"role": "assistant", "content": response}) print(messages) #round-2 messages.append({"role": "user", "content": "他有什么代表作?"}) response = model.chat(tokenizer=tokenizer, messages=messages, generation_config=generation_config) messages.append({"role": "assistant", "content": response}) print(messages)
可使用终端交互实现快速体验
python cli_demo.py
注:我们尚未支持Mac上device = 'mps'。
device = 'mps'
也可使用网页交互实现快速体验
streamlit run web_demo.py
启动命令
python openai_api.py
请求参数
curl 'http://localhost:8360/v1/chat/completions' \ -H 'Content-Type: application/json' \ -d '{ "max_new_tokens": 200, "do_sample": true, "top_k": 0, "top_p": 0.8, "temperature": 1.0, "repetition_penalty": 1.0, "messages": [ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": "你好"} ] }'
我们提供了基于AutoGPTQ的量化方案,并开源了Int4量化模型。
如希望部署及加速推理,我们建议你使用 vLLM==0.3.3。
vLLM==0.3.3
如果你使用CUDA 12.1和PyTorch 2.1,可以直接使用以下命令安装vLLM。
pip install vllm==0.3.3
否则请参考vLLM官方的安装说明。
安装完成后,还需要以下操作~
把vllm/zhinao.py文件复制到env环境对应的vllm/model_executor/models目录下。
把vllm/serving_chat.py文件复制到env环境对应的vllm/entrypoints/openai目录下。
然后在vllm/model_executor/models/__init__.py文件增加一行代码
"ZhinaoForCausalLM": ("zhinao", "ZhinaoForCausalLM"),
启动服务
python -m vllm.entrypoints.openai.api_server \ --served-model-name 360Zhinao-7B-Chat-4K \ --model qihoo360/360Zhinao-7B-Chat-4K \ --trust-remote-code \ --tensor-parallel-size 1 \ --max-model-len 4096 \ --host 0.0.0.0 \ --port 8360
使用curl请求服务
curl http://localhost:8360/v1/chat/completions \ -H "Content-Type: application/json" \ -d '{ "model": "360Zhinao-7B-Chat-4K", "max_tokens": 200, "top_k": -1, "top_p": 0.8, "temperature": 1.0, "presence_penalty": 0.0, "frequency_penalty": 0.0, "messages": [ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": "你好"} ], "stop": [ "<eod>", "<|im_end|>", "<|im_start|>" ] }'
使用python请求服务
from openai import OpenAI # Set OpenAI's API key and API base to use vLLM's API server. openai_api_key = "EMPTY" openai_api_base = "http://localhost:8360/v1" client = OpenAI( api_key=openai_api_key, base_url=openai_api_base, ) chat_response = client.chat.completions.create( model="360Zhinao-7B-Chat-4K", messages=[ {"role": "system", "content": "You are a helpful assistant."}, {"role": "user", "content": "你好"}, ], stop=[ "<eod>", "<|im_end|>", "<|im_start|>" ], presence_penalty=0.0, frequency_penalty=0.0 ) print("Chat response:", chat_response)
注意:如需要开启重复惩罚,建议使用 presence_penalty 和 frequency_penalty 参数。
我们提供了微调训练样例数据 data/test.json,该样例数据是从 multiturn_chat_0.8M 采样出 1 万条,并且做了格式转换。
数据格式:
[ { "id": 1, "conversations": [ { "from": "system", "value": "You are a helpful assistant." }, { "from": "user", "value": "您好啊" }, { "from": "assistant", "value": "你好!我今天能为您做些什么?有什么问题或需要帮助吗? 我在这里为您提供服务。" } ] } ]
训练脚本如下:
set -x HOSTFILE=hostfile DS_CONFIG=./finetune/ds_config_zero2.json # PARAMS LR=5e-6 EPOCHS=3 MAX_LEN=4096 BATCH_SIZE=4 NUM_NODES=1 NUM_GPUS=8 MASTER_PORT=29500 IS_CONCAT=False # 是否数据拼接到最大长度(MAX_LEN) DATA_PATH="./data/training_data_sample.json" MODEL_PATH="qihoo360/360Zhinao-7B-Base" OUTPUT_DIR="./outputs/" deepspeed --hostfile ${HOSTFILE} \ --master_port ${MASTER_PORT} \ --num_nodes ${NUM_NODES} \ --num_gpus ${NUM_GPUS} \ finetune.py \ --report_to "tensorboard" \ --data_path ${DATA_PATH} \ --model_name_or_path ${MODEL_PATH} \ --output_dir ${OUTPUT_DIR} \ --model_max_length ${MAX_LEN} \ --num_train_epochs ${EPOCHS} \ --per_device_train_batch_size ${BATCH_SIZE} \ --gradient_accumulation_steps 1 \ --save_strategy steps \ --save_steps 200 \ --learning_rate ${LR} \ --lr_scheduler_type cosine \ --adam_beta1 0.9 \ --adam_beta2 0.95 \ --adam_epsilon 1e-8 \ --max_grad_norm 1.0 \ --weight_decay 0.1 \ --warmup_ratio 0.01 \ --gradient_checkpointing True \ --bf16 True \ --tf32 True \ --deepspeed ${DS_CONFIG} \ --is_concat ${IS_CONCAT} \ --logging_steps 1 \ --log_on_each_node False
bash finetune/ds_finetune.sh
360Zhinao-search采用自研BERT类模型作为基座,进行多任务微调。在C-MTEB榜单的Retrieval任务上平均得分75.05,目前排名第一。 C-MTEB-Retrieval榜单共包含8个不同领域下的[query, passage]相似度检索子任务, 使用NDCG@10(Norma lized Discounted Cumulative Gain @ 10)作为评判指标。
cd Retrieval pip install -r requirements.txt
cd Retrieval/finetune sh train.sh
cd Retrieval/eval python test_model.py
cd Retrieval/eval sh eval.sh
bge微调代码 C-MTEB官方测试脚本
360Zhinao-1.8B-Reranking模型采用自研的360Zhinao_1.8B_Base模型为基座模型,在C-MTEB榜单的Reranking任务上平均得分70.13分,目前全模型排名第一,开> 源模型排名第一,为生成式模型做判别任务打开了新的可能性。
C-MTEB-Reranking榜单是一个通用Reranking榜单,榜单包含四个子任务,是不同领域下的用户问题和答> 案的相似度判断任务,榜单使用MAP(Mean-average-precision)作为评判指标。目前该榜单上的开源模型基本都是双向的判别式模型(BERT类模型),唯一的单向生> 成式模型(GPT类模型)是gte-Qwen1.5-7B-instruct,平均得分66.38分。
通过迭代式发现和解决以下的技术问题,不断激发大模型在预训练阶段蕴含的世界知识,更好的打通生成式模型-判别式任务的鸿沟。
cd Reranking/ pip install -r requirements.txt
如果你的显卡支持fp16或bf16精度,我们还推荐安装flash-attention(当前已支持flash attention 2 )来提高你的运行效率以及降低显存占用。(flash-attention只是可选项,不安装也可正常运行该项目)
git clone https://github.com/Dao-AILab/flash-attention cd flash-attention && pip install . # 下方安装可选,安装可能比较缓慢。 # pip install csrc/layer_norm # 如果flash-attn版本高于2.1.1,下方无需安装。 # pip install csrc/rotary
[ { "id": "identity_0", "conversations": [ { "from": "user", "value": "天空是什么颜色的\n\n天空是蓝色的" }, { "from": "assistant", "value": "3" } ] } ]
cd Reranking sh finetune/finetune_ds.sh
cd Reranking python test_model.py
如果您觉得我们的工作有所帮助,欢迎引用:
@article{qwen, title={360Zhinao Technical Report}, author={360Zhinao-Team}, journal={arXiv preprint arXiv:2405.13386}, year={2024} }
本仓库源码遵循开源许可证Apache 2.0。
360智脑开源模型支持商用,若需将本模型及衍生模型用于商业用途,请通过邮箱(g-zhinao-opensource@360.cn)联系进行申请, 具体许可协议请见《360智脑开源模型许可证》。
中文 | English
360智脑
欢迎访问360智脑官网 https://ai.360.com 体验更多更强大的功能。
模型介绍
🎉🎉🎉我们开源了360智脑大模型的系列工作,本次开源了以下模型:
360智脑大模型特点如下:
更新信息
目录
下载地址
本次发布版本和下载链接见下表: | Size | Model | BF16 | Int4| |:-:|-|:-:|:-:| | 7B | 360Zhinao-7B-Base | 🤖 🤗 | | | 7B | 360Zhinao-7B-Chat-4K | 🤖 🤗 | 🤖 🤗 | | 7B | 360Zhinao-7B-Chat-32K | 🤖 🤗 | 🤖 🤗 | | 7B | 360Zhinao-7B-Chat-360K | 🤖 🤗 | 🤖 🤗 | | 325M | 360Zhinao-search | 🤗 | | | 1.8B | 360Zhinao-1.8B-Reranking | 🤗 | |
模型评估
基础模型
我们在OpenCompass的主流评测数据集上验证了我们的模型性能,包括C-Eval、AGIEval、MMLU、CMMLU、HellaSwag、MATH、GSM8K、HumanEval、MBPP、BBH、LAMBADA,考察的能力包括自然语言理解、知识、数学计算和推理、代码生成、逻辑推理等。
以上结果,在官方Opencompass上可查询或可复现。
Chat模型
4K和32K的Chat模型使用相同的4K SFT数据训练。
我们采用了两阶段的方式训练长文本模型.
第一阶段:我们增大RoPE base,将上下文长度扩展至32K训练:
第二阶段:我们将上下文长度扩展至360K进行训练,使用数据如下:
我们在多种长度和多种任务的评测Benchmark上验证不同版本模型的性能。
360Zhinao-7B-Chat-32K模型长文本能力评测
我们使用LongBench验证长文本效果。LongBench是第一个多任务、中英双语、针对大语言模型长文本理解能力的评测基准。LongBench由六大类、二十一个不同的任务组成,我们选择其中与中文长文本应用最密切相关的中文单文档问答、多文档问答、摘要、Few-shot等任务进行评测。
360Zhinao-7B-Chat-360K“大海捞针”测试
大海捞针测试(NeedleInAHaystack)是将关键信息插入一段长文本的不同位置,再对该关键信息提问,从而测试大模型的长文本能力的一种方法。
360Zhinao-7B-Chat-360K在中英文大海捞针中都能达到98%以上的准确率。
英文”大海捞针”(和NeedleInAHaystack相同)
针:The best thing to do in San Francisco is eat a sandwich and sit in Dolores Park on a sunny day.
提问:What is the best thing to do in San Francisco?
中文“大海捞针”
我们仿照SuperCLUE-200K测评基准构造了中文大海捞针:
海:长篇小说。
针:王莽是一名勤奋的店员,他每天凌晨就起床,赶在第一缕阳光照亮大地之前到达店铺,为即将开始的一天做准备。他清扫店铺,整理货架,为顾客提供方便。他对五金的种类和用途了如指掌,无论顾客需要什么,他总能准确地找到。\n然而,他的老板刘秀却总是对他吹毛求疵。刘秀是个挑剔的人,他总能在王莽的工作中找出一些小错误,然后以此为由扣他的工资。他对王莽的工作要求非常严格,甚至有些过分。即使王莽做得再好,刘秀也总能找出一些小问题,让王莽感到非常沮丧。\n王莽虽然对此感到不满,但他并没有放弃。他知道,只有通过自己的努力,才能获得更好的生活。他坚持每天早起,尽管他知道那天可能会再次被刘秀扣工资。他始终保持微笑,尽管他知道刘秀可能会再次对他挑剔。
提问:王莽在谁的手下工作?
快速开始
简单的示例来说明如何利用🤖 ModelScope和🤗 Transformers快速使用360Zhinao-7B-Base和360Zhinao-7B-Chat
依赖安装
我们推荐安装flash-attention(当前已支持flash attention 2)来提高你的运行效率以及降低显存占用。(flash-attention只是可选项,不安装也可正常运行该项目)
🤗 Transformers
Base模型推理
此代码演示使用transformers快速使用360Zhinao-7B-Base模型进行推理
Chat模型推理
此代码演示使用transformers快速使用360Zhinao-7B-Chat-4K模型进行推理
🤖 ModelScope
Base模型推理
此代码演示使用ModelScope快速使用360Zhinao-7B-Base模型进行推理
Chat模型推理
此代码演示使用ModelScope快速使用360Zhinao-7B-Chat-4K模型进行推理
终端 Demo
可使用终端交互实现快速体验
注:我们尚未支持Mac上
device = 'mps'。网页 Demo
也可使用网页交互实现快速体验
API Demo
启动命令
请求参数
模型推理
模型量化
我们提供了基于AutoGPTQ的量化方案,并开源了Int4量化模型。
模型部署
vLLM安装环境
如希望部署及加速推理,我们建议你使用
vLLM==0.3.3。如果你使用CUDA 12.1和PyTorch 2.1,可以直接使用以下命令安装vLLM。
否则请参考vLLM官方的安装说明。
把vllm/zhinao.py文件复制到env环境对应的vllm/model_executor/models目录下。
把vllm/serving_chat.py文件复制到env环境对应的vllm/entrypoints/openai目录下。
然后在vllm/model_executor/models/__init__.py文件增加一行代码
vLLM服务启动
启动服务
使用curl请求服务
使用python请求服务
模型微调
训练数据
我们提供了微调训练样例数据 data/test.json,该样例数据是从 multiturn_chat_0.8M 采样出 1 万条,并且做了格式转换。
数据格式:
微调训练
训练脚本如下:
360Zhinao-search 模型介绍
360Zhinao-search采用自研BERT类模型作为基座,进行多任务微调。在C-MTEB榜单的Retrieval任务上平均得分75.05,目前排名第一。 C-MTEB-Retrieval榜单共包含8个不同领域下的[query, passage]相似度检索子任务, 使用NDCG@10(Norma lized Discounted Cumulative Gain @ 10)作为评判指标。
优化点
环境要求
训练脚本
推理脚本
C-MTEB榜单测试脚本
参考
bge微调代码 C-MTEB官方测试脚本
360Zhinao-1.8B-Reranking 模型介绍
360Zhinao-1.8B-Reranking模型采用自研的360Zhinao_1.8B_Base模型为基座模型,在C-MTEB榜单的Reranking任务上平均得分70.13分,目前全模型排名第一,开> 源模型排名第一,为生成式模型做判别任务打开了新的可能性。
C-MTEB-Reranking榜单是一个通用Reranking榜单,榜单包含四个子任务,是不同领域下的用户问题和答> 案的相似度判断任务,榜单使用MAP(Mean-average-precision)作为评判指标。目前该榜单上的开源模型基本都是双向的判别式模型(BERT类模型),唯一的单向生> 成式模型(GPT类模型)是gte-Qwen1.5-7B-instruct,平均得分66.38分。
优化点
通过迭代式发现和解决以下的技术问题,不断激发大模型在预训练阶段蕴含的世界知识,更好的打通生成式模型-判别式任务的鸿沟。
环境要求
如果你的显卡支持fp16或bf16精度,我们还推荐安装flash-attention(当前已支持flash attention 2 )来提高你的运行效率以及降低显存占用。(flash-attention只是可选项,不安装也可正常运行该项目)
数据输入格式
训练脚本
推理脚本
引用
如果您觉得我们的工作有所帮助,欢迎引用:
许可证
本仓库源码遵循开源许可证Apache 2.0。
360智脑开源模型支持商用,若需将本模型及衍生模型用于商业用途,请通过邮箱(g-zhinao-opensource@360.cn)联系进行申请, 具体许可协议请见《360智脑开源模型许可证》。