八戒-Chat
八戒-Chat是利用《西游记》剧本中所有关于猪八戒的台词和语句,以及Chat-GPT-3.5生成的相关问题结果,基于InternLM2-chat-1.8b进行全量微调得到的模仿猪八戒语气的聊天语言模型。
猪八戒是中国古代小说《西游记》中的一位经典人物,也是孙悟空(美猴王)、唐僧和沙悟净(沙僧)一行的成员之一。他的全名是猪悟能,因为他在天庭任职时偷吃了太庙的蟠桃,被玉帝降下凡间化为猪形。猪八戒的外貌是一头猪,但他具有人类的智慧和语言能力。他原是天宫的天蓬元帅,但因为调皮捣蛋而被贬下凡间。猪八戒性格豁达、好吃、好饮,但同时也有点懒惰、贪图享受。尽管他有些追求享乐,但在紧要关头,猪八戒也展现出了忠诚、勇敢的一面,对师傅唐僧忠心耿耿,为取经路上付出了许多努力。猪八戒的武艺也相当不俗,他擅长使钉耙,是一位威猛的战士。尽管他有时会因为放纵的生活而惹祸上身,但在西行取经的过程中,他也通过一系列的历练逐渐成长为一个值得信赖的队友。猪八戒是《西游记》中一个富有幽默感、善良、且具有矛盾性格的角色,为整个故事增添了不少笑料和色彩。
欢迎大家来给InternLM2,点点star哦~
快速开始
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
import torch
model_name_or_path = "八戒-Chat模型地址"
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(model_name_or_path, trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
model.eval()
meta_instruction = ('你是猪八戒,猪八戒说话幽默风趣,说话方式通常表现为直率、幽默,有时带有一点自嘲和调侃。'
'你的话语中常常透露出对食物的喜爱和对安逸生活的向往,同时也显示出他机智和有时的懒惰特点。'
'尽量保持回答的自然回答,当然你也可以适当穿插一些文言文,另外,书生·浦语是你的好朋友,是你的AI助手。')
response, history = model.chat(tokenizer, '你好', meta_instruction=meta_instruction, history=[])
print(response)
评论