介绍
CharacterGLM-6B 是 聆心智能和清华大学 CoAI 实验室联合发布的新一代对话预训练模型。CharacterGLM-6B 是 基于 ChatGLM2 系列中的开源模型,在保留了前两代模型对话流畅、部署门槛低等众多优秀特性的基础上,CharacterGLM-6B 的设计遵循以下原则:
AI角色变“活”的强心针
一个对话式的AI角色要想表现的像一个栩栩如生的人,必定需要具备“人的特质”,特别是人在语言表达中的文本形式的特征。我们将人的语言表达特征的重点落实在属性和行为上:属性主要影响语言表达的内容,行为则影响语言表达的风格和口吻。
属性: CharacterGLM的设计主要考虑了七种属性,包括身份、兴趣、观点、经历、成就、社交关系和其他。
行为: 行为主要由一些动态的元素组成:语言特征、情感表达和互动模式。例如,老年人更倾向于使用一些更正式的语言,而青少年则更喜欢用网络流行语。CharacterGLM则主要考虑了语言学特征和性格作为行为方面的设计。
AI角色是否“活”的照妖镜
一个对话式的AI角色要想证明自己是一个栩栩如生的角色,需要具备真实的人所具备的表达特质。我们主要关注三个方面的表达特质:一致性、拟人化和吸引力。
一致性属性: 角色一致性是角色在交互期间展现稳定的属性和行为的能力。维持一个会话式AI角色在对话中属性和行为的一致对于赢得用户的满足和信任是至关重要的。
拟人化行为: 角色拟人化要求角色在与用户的交互中表现自然,类似人与人之间的自然交互。类人的会话式AI角色对于提高用户的接受度以及促进更自然和有吸引力的对话是不可或缺的。
吸引力: 吸引力是会话式AI角色引起用户兴趣以及促进用户参与的衡量依据。聊天过程中,让对话有趣,让人想聊下去会直接影响用户的体验,这也是对话模型整体性能的一个体现。
错误分析(如何评估数字人)
我们对11个模型每个轮次的回复进一步标注了六个方面:角色不一致(OOC)、矛盾(Contradiction)、重复(Repetition)、低质量(Less-quality)、低信息量(Less-information)和主动性(Proactivity,主动引导话题并推动对话发展的能力)。此外,“整体(Overall)”分数的计算方式为前五个维度的总和减去第六个维度,“整体”得分越低表示性能越好。
对比式评估
我们将 CharacterGLM 与专门用于角色扮演的 MiniMax 模型以及 GPT-3.5 和 GPT-4进行了对比式的评估。该评估仍采用交互式人工评估,共涉及24个角色,涵盖名人类、日常生活类、游戏影音类以及虚拟恋爱类角色,对话主题限制在闲聊、访谈和恋爱三种场景。同样地,这里雇佣了10个标注人员与模型进行交互,并标记两个模型在相同上下文下的两个输出为胜(win)/平(tie)/负(lose),最终计算每个模型在不同角色类别和对话主题下的胜/平/负比率。
使用方式
环境安装
首先需要下载本仓库:
git clone https://github.com/thu-coai/CharacterGLM-6B
cd CharacterGLM-6b
然后使用 pip 安装依赖:
pip install -r requirements.txt
transformers
库版本应该4.36.2
以及以上的版本 ,torch
库版本应为 2.1.0 及以上的版本,以获得最佳的推理性能。- 为了保证
torch
的版本正确,请严格按照 官方文档 的说明安装。
从本地加载模型
自动下载模型实现和参数。完整的模型实现在 Hugging Face Hub 。如果你的网络环境较差,下载模型参数可能会花费较长时间甚至失败。此时可以先将模型下载到本地,然后从本地加载。
从 Hugging Face Hub 下载模型需要先安装Git LFS ,然后运行
git lfs install
git clone https://huggingface.co/thu-coai/CharacterGLM-6B
网页版对话 Demo
可以通过以下命令启动基于 Streamlit 的网页版 demo:
一定要进入文件夹运行,防止出现无法找到character.json
的问题
cd basic_demo
streamlit run web_demo_streamlit.py
网页版 demo 会运行一个 Web Server,并输出地址。在浏览器中打开输出的地址即可使用。 经测试,基于 Streamlit 的网页版 Demo 会更流畅。
命令行对话 Demo
运行仓库中 cli_demo.py:
python basic_demo/cli_demo.py
程序会在命令行中进行交互式的对话,在命令行中输入指示并回车即可生成回复,输入 clear
可以清空对话历史,输入 stop
终止程序。
自此,全部结束
参考Character.AI:个性化的ChatGPT,AI大模型时代的UGC平台 | 人人都是产品经理 (woshipm.com)