《AI大模型应用》-StyleLLM文风大模型：基于大语言模型的文本风格迁移项目.zip

<div align="center"> <h1>StyleLLM 文风大模型</h1> </div> # 模型介绍 **stylellm** 是一个基于大语言模型（llm）的文本风格迁移（text style transfer）项目。项目利用大语言模型来学习指定文学作品的写作风格（惯用词汇、句式结构、修辞手法、人物对话等），形成了一系列特定风格的模型。 利用**stylellm**模型可将学习到的风格移植至其他通用文本上，即：输入一段原始文本，模型可对其改写，输出带有该风格特色的文本，达到文字修饰、润色或风格模仿的效果。 # 项目内容 - 📚 [四大名著风格模型](#模型列表) ([量化版本](#%E9%87%8F%E5%8C%96%E6%A8%A1%E5%9E%8B%E5%88%97%E8%A1%A8awq4bits)) - ↔️ [风格转化效果](#风格转化效果) - 💬 [风格化聊天效果](#风格化聊天效果) # 模型列表 **stylellm-中国四大名著系列** 包括基于Yi-6b微调的四个模型，分别采用《三国演义》、《西游记》、《水浒传》、《红楼梦》四本中国古典长篇小说训练而成： | 风格 | 模型 | 使用方法 | |:----------|:----------|:----------| | 三国演义风格 | 🤗 [stylellm/SanGuoYanYi-6b](https://huggingface.co/stylellm/SanGuoYanYi-6b) | [inference.py](https://github.com/stylellm/stylellm_models/blob/main/examples/SanGuoYanYi-6b/inference.py) | | 西游记风格 | 🤗 [stylellm/XiYouJi-6b](https://huggingface.co/stylellm/XiYouJi-6b) | [inference.py](https://github.com/stylellm/stylellm_models/blob/main/examples/XiYouJi-6b/inference.py) | | 水浒传风格 | 🤗 [stylellm/ShuiHuZhuan-6b](https://huggingface.co/stylellm/ShuiHuZhuan-6b) | [inference.py](https://github.com/stylellm/stylellm_models/blob/main/examples/ShuiHuZhuan-6b/inference.py) | | 红楼梦风格 | 🤗 [stylellm/HongLouMeng-6b](https://huggingface.co/stylellm/HongLouMeng-6b) | [inference.py](https://github.com/stylellm/stylellm_models/blob/main/examples/HongLouMeng-6b/inference.py) | <br/> ## 量化模型列表（AWQ，4bits） | 风格 | 模型 | |:----------|:----------| | 三国演义风格 | 🤗 [stylellm/SanGuoYanYi-6b-AWQ](https://huggingface.co/stylellm/SanGuoYanYi-6b-AWQ) | | 西游记风格 | 🤗 [stylellm/XiYouJi-6b-AWQ](https://huggingface.co/stylellm/XiYouJi-6b-AWQ) | | 水浒传风格 | 🤗 [stylellm/ShuiHuZhuan-6b-AWQ](https://huggingface.co/stylellm/ShuiHuZhuan-6b-AWQ) | | 红楼梦风格 | 🤗 [stylellm/HongLouMeng-6b-AWQ](https://huggingface.co/stylellm/HongLouMeng-6b-AWQ) | > **注意：**<br/> > 使用AWQ量化模型前需安装[AutoAWQ](https://github.com/casper-hansen/AutoAWQ)：pip install autoawq <br/> > 量化后的模型占用显存4.2GB，适用于较小的显卡使用，但风格表达能力有所下降，追求完整效果的场景请使用[量化前版本](#模型列表) # 风格转化效果 以《白雪公主》原文作为输入，以下是不同风格模型的改写结果。 | 原文 - 《白雪公主》 | 改写-三国演义风格 | 改写-西游记风格 | 改写-水浒传风格 | 改写-红楼梦风格 | |:----------|:----------|:----------|:----------|:----------| | 严冬时节，鹅毛一样的大雪片在天空中到处飞舞着，有一个王后坐在王宫里的一扇窗子边，正在为她的女儿做针线活儿，寒风卷着雪片飘进了窗子，乌木窗台上飘落了不少雪花。 | 时值隆冬，大雪压境，王后坐于王宫之窗下，正与女作绣，寒风入室，吹落乌木窗台之雪。 | 正是那：朔风吹凛凛，瑞雪霭纷纷；王后坐窗前，正教女作绣，只见那：寒风送雪入窗棂，乌木台头堆积雪。 | 此时正是隆冬之际，鹅毛般大雪乱飘，有一王后坐于王宫内一扇窗户前，正缝其女衣，寒风卷雪入得窗来，把那乌木窗台都压坏了。 | 且说那严冬天气，鹅毛大雪，纷纷扬扬，只见那王后坐于王宫之窗下，正教他女儿作针黹，忽见寒风卷雪，吹入窗内，乌木窗台之上，积了半日之雪。 | | 她抬头向窗外望去，一不留神，针刺进了她的手指，红红的鲜血从针口流了出来，有三点血滴落在飘进窗子的雪花上。 | 其女仰视窗外，不料针落指中，遂出血于雪，有三点之痕。 | 只见那女子举目望外观看时，不期把个指头儿扎了一针，又见出血，却似三两点血珠儿落于雪内。 | 那娘子仰面看时，只道是雪片落将下来，不提防把针儿捻在手里，一时失手，正中指尖，便见出血来，有三两点，都滴入窗里去。 | 只见那丫头往外瞧时，一时针扎手了，便出血了，有三四个血点儿，落在了雪里。 | | 她若有所思地凝视着点缀在白雪上的鲜红血滴，又看了看乌木窗台，说道："但愿我小女儿的皮肤长得白里透红，看起来就像这洁白的雪和鲜红的血一样，那么艳丽，那么骄嫩，头发长得就像这窗子的乌木一般又黑又亮！" | 其女乃凝眸望那点缀于白雪之下的鲜红血滴，复观乌木窗棂曰：“吾愿吾小女肌肤如雪，容貌似此洁白之雪与鲜红之血，娇嫩之发，正如此窗棂之乌木也。” | 只见那般血迹，却也殷红；又见那般白雪，却也莹洁。她又望了望那乌木窗棂道：“但愿吾女肌肤如雪，颜色似朱，发髻如漆，方才好哩！” | 只见那女子把眼来望那白雪上点点鲜血，又看那乌木窗子道："只愿俺的小女儿皮肉也似这般白净，便如这雪相似；须发也似这般漆黑光鲜，便似这乌木般好！" | 一面看那白雪中点点血痕，一面又瞧那乌木窗棂，因叹道：“只望我小女肌肤丰泽，如这雪白，血红，发也似的乌木一般。” | | 她的小女儿渐渐长大了，小姑娘长得水灵灵的，真是人见人爱，美丽动人。她的皮肤真的就像雪一样的白嫩，又透着血一样的红润，头发像乌木一样的黑亮。所以王后给她取了个名字，叫白雪公主。 | 其女渐长，乃名曰：“白雪”，其色如雪，真美之极。肤若冰雪，发似乌木，故王后以白雪为号。 | 光阴似箭，不觉数载，那小女儿渐长成人，生得娉婷娇艳，真个是：肤如凝脂，貌若桃李，发似漆黑，故此王后名曰白雪公主。 | 那小女儿渐长成，生得十分美貌，真个有倾国之色，娇艳无伦。肌肤如玉，面似桃花，发若漆黑。因此王后便取名唤做"白雪公主"。 | 那丫头渐长，生得袅娜纤巧，乖觉伶俐，肌肤丰泽，容貌娇美，且是雪肌花颜，唇若施脂，腰如束柳。故此，王后便命名曰：“白雪”。 | | 但白雪公主还没有长大，她的王后妈妈就死去了。不久，国王爸爸又娶了一个妻子。这个王后长得非常漂亮，但她很骄傲自负，嫉妒心极强，只要听说有人比她漂亮，她都不能忍受。 | 却说白雪公主年方幼小，其母王后乃亡矣。不数日，父王复纳新妇，此妇美而骄妒，闻人言有胜己者，不能容也。 | 那白雪公主尚幼年，其母王后已亡矣。不多时，国王复纳一妻，此女貌美而骄矜，妒忌之心甚重，闻得人言有胜于己者，即不能容也。 | 此时白雪公主尚幼小，其母王后便亡故了；不数日间，父王复纳新妇为妻。此女美貌绝伦，性傲气高，妒忌之心甚重，闻有胜己者，便不能容。 | 且说那白雪公主尚未长成，其母王后已亡故了。不几日，国王复纳一妻，此妃生得甚美，惟性傲慢，妒忌之心颇重，凡见人胜过己者，便不能容。 | | 她有一块魔镜，她经常走到镜子面前自我欣赏，并问道："告诉我，镜子，告诉我实话！这儿所有的女人谁最漂亮？告诉我她是谁？"镜子回答道："是你，王后！你就是这儿最漂亮的女人。"听到这样的话，她就会满意地笑起来。 | 王后乃有魔镜，常自照其