/Zhiyin

Primary LanguagePython

Zhiyin

Zhiyin-Chat

知音聊天机器人由中科院声学所语音与智能信息处理实验室研发,用于回答用户问题和提供信息,以帮助人们解决问题和获取知识。本项目后续会开放指令训练数据、相关模型、训练代码、应用场景等。

📝 项目主要内容

🚀 推理代码

详见cli_demo.py,尽可能简化的一个推理代码实现。

🤖 模型

详见Huggingface , 我们实验室研发的大模型Zhiyin-7B-Chat

安装依赖

pip install -r requirements.txt

Python使用范例

>>> import torch
>>> from transformers.generation.utils import GenerationConfig
>>> from transformers import AutoModelForCausalLM, AutoTokenizer
>>> model_dir='HCCL-NLP/Zhiyin-7B-Chat'#离线使用时改为模型储存路径
>>> tokenizer = AutoTokenizer.from_pretrained(model_dir, use_fast=False, trust_remote_code=True)
>>> model = AutoModelForCausalLM.from_pretrained(model_dir, device_map='auto', torch_dtype=torch.bfloat16, trust_remote_code=True)
>>> model.generation_config = GenerationConfig.from_pretrained(model_dir)
>>> messages = []
>>> messages.append({"role": "user", "content": "诸葛亮北伐失败的原因"})
>>> response = model.chat(tokenizer, messages)
>>> print(response)
诸葛亮北伐失败的原因有很多以下是一些主要因素1. 资源不足尽管蜀国在诸葛亮的领导下进行了积极的备战但与强大的魏国相比其资源和实力仍然相差甚远2. 地理劣势魏国占据着中原地区地势平坦易守难攻而蜀国则地处西南地势险要但运输困难3. 战略失误诸葛亮北伐时曾试图攻打魏国的首都洛阳但这一战略被许多历史学家认为是错误的因为这样会陷入敌人的腹地难以持4. 人才流失蜀国在诸葛亮北伐期间人才流失严重尤其是蜀汉的精英将领和士兵这大大削弱了蜀国的实力5. 时间有限诸葛亮北伐的时间有限他在五丈原与司马懿相持数月后因病去世北伐计划被迫中止6. 内部矛盾蜀国内部存在一定的矛盾和冲突如诸葛亮与李严的关系紧张等这些矛盾对北伐产生了一定的影响