CharacterGLM-6B的搭建

介绍

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)

  • 19
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值