license: other license_name: yi-license license_link: https://huggingface.co/01-ai/Yi-6B/blob/main/LICENSE language:

• zh tags:
• 文本风格迁移
• 文学

模型介绍

stylellm 是一个基于大语言模型（LLM）的文本风格迁移项目。该项目利用大语言模型学习指定文学作品的写作风格（包括惯用词汇、句式结构、修辞手法、人物对话等），形成了一系列特定风格的模型。

通过stylellm模型，可以将学习到的文学风格应用于其他通用文本：输入原始文本后，模型会进行改写，输出带有该风格特色的文本，实现文字修饰、润色或风格模仿的效果。

模型列表

stylellm-中国四大名著系列包含基于Yi-6b微调的四个模型，分别采用《三国演义》、《西游记》、《水浒传》、《红楼梦》四部中国古典长篇小说训练而成：

效果演示

以《白雪公主》原文为输入，展示不同风格模型的改写效果。

使用方法

from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("stylellm/SanGuoYanYi-6b")
model = AutoModelForCausalLM.from_pretrained("stylellm/SanGuoYanYi-6b").eval()

messages = [{"role": "user", "content": "严冬时节，鹅毛一样的大雪片在天空中到处飞舞着，有一个王后坐在王宫里的一扇窗子边，正在为她的女儿做针线活儿，寒风卷着雪片飘进了窗子，乌木窗台上飘落了不少雪花。"}]
input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
output_ids = model.generate(input_ids, do_sample=False, repetition_penalty=1.2)
response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
print("Output:", response)
# 输出: 时值隆冬，大雪压境，王后坐于王宫之窗下，正与女作绣，寒风入室，吹落乌木窗台之雪。

致谢

本项目在模型训练中使用了以下开源项目，特此致谢：

• 01-ai/Yi: 本项目模型底座
• hiyouga/LLaMA-Factory: 本项目训练框架