Update README.md
Browse files
README.md
CHANGED
@@ -7,4 +7,215 @@ sdk: static
|
|
7 |
pinned: false
|
8 |
---
|
9 |
|
10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
pinned: false
|
8 |
---
|
9 |
|
10 |
+
<!-- markdownlint-disable first-line-h1 -->
|
11 |
+
<!-- markdownlint-disable html -->
|
12 |
+
<div align="center">
|
13 |
+
<h1>
|
14 |
+
Sakura-13B-Galgame
|
15 |
+
</h1>
|
16 |
+
</div>
|
17 |
+
|
18 |
+
<p align="center">
|
19 |
+
🦉 <a href="https://github.com/pipixia244/Sakura-13B-Galgame" target="_blank">GitHub</a> | 🤖 <a href="https://www.modelscope.cn/models/sakuraumi/Sakura-13B-Galgame" target="_blank">ModelScope</a>
|
20 |
+
</p>
|
21 |
+
|
22 |
+
# 介绍
|
23 |
+
|
24 |
+
基于LLaMA2-13B,OpenBuddy(v0.1-v0.4)和Baichuan2-13B(v0.5+)构建,在Galgame中日文本数据上进行微调,旨在提供性能接近GPT3.5且完全离线的Galgame/轻小说翻译大语言模型. 新建了[TG交流群](https://t.me/+sCYaCYEsd3ZkMTE1),欢迎交流讨论。
|
25 |
+
|
26 |
+
模型下载:
|
27 |
+
| 版本 | 全量模型 | 8-bit量化 | 4-bit量化|
|
28 |
+
|:-------:|:-------:|:-------:|:-------:|
|
29 |
+
| 20230827-v0.1 | 🤗 [Sakura-13B-Galgame-v0.1](https://huggingface.co/sakuraumi/Sakura-13B-Galgame/tree/main/sakura_13b_model_v0.1) | - | - |
|
30 |
+
| 20230908-v0.4 | 🤗 [Sakura-13B-Galgame-v0.4](https://huggingface.co/sakuraumi/Sakura-13B-Galgame/tree/main/sakura_13b_model_v0.4) | - | - |
|
31 |
+
| 20230917-v0.5 | 🤗 sakuraumi/Sakura-13B-Galgame根目录下 | 🤗 [Sakura-13B-Galgame-v0.5-8bits](https://huggingface.co/sakuraumi/Sakura-13B-Galgame/tree/main/sakura_13b_model_v0.5_8bits) | [Sakura-13B-Galgame-v0.5-4bits](https://huggingface.co/sakuraumi/Sakura-13B-Galgame/tree/main/sakura_13b_model_v0.5_4bits_autogptq_40k) |
|
32 |
+
| 20231011-v0.7 | 🤗 [Sakura-14B-LNovel](https://huggingface.co/sakuraumi/Sakura-14B-LNovel)
|
33 |
+
|
34 |
+
目前仍为实验版本,翻译质量较差.
|
35 |
+
|
36 |
+
~~lsp们最关注的翻译R18轻小说的效果:使用[这本r18小说](https://novel18.syosetu.com/n2975ig)的部分文本,比较各版本输出文本(其中GPT网页端和API均无法翻译),结果已上传至[链接](https://huggingface.co/sakuraumi/Sakura-13B-Galgame-Archived/blob/main/r18_test.txt)~~
|
37 |
+
|
38 |
+
|
39 |
+
# 日志
|
40 |
+
|
41 |
+
`20231011`:上传第四版模型`sakura-14b-2epoch-4.4M-1003-v0.7`,改用QWen-14B-Chat模型进行微调,增加数据集。
|
42 |
+
|
43 |
+
`20230917`:上传第三版模型`sakura-13b-2epoch-2.6M-0917-v0.5`,改用Baichuan2-13B-Chat模型进行微调,翻译质量有所提高。
|
44 |
+
|
45 |
+
`20230908`:上传第二版模型`sakura-13b-1epoch-2.6M-0903-v0.4`,使用Galgame和轻小说数据集进行微调,语法能力有所提高。轻小说数据集感谢[CjangCjengh](https://github.com/CjangCjengh)大佬提供。
|
46 |
+
|
47 |
+
`20230827`:上传第一版模型`sakura-13b-2epoch-260k-0826-v0.1`
|
48 |
+
|
49 |
+
# 模型详情
|
50 |
+
|
51 |
+
## 描述
|
52 |
+
|
53 |
+
### v0.1-v0.4
|
54 |
+
|
55 |
+
- Finetuned by [SakuraUmi](https://github.com/pipixia244)
|
56 |
+
- Finetuned on [Openbuddy-LLaMA2-13B](https://huggingface.co/OpenBuddy/openbuddy-llama2-13b-v8.1-fp16)
|
57 |
+
- Base model: [LLaMA2-13B](https://huggingface.co/meta-llama/Llama-2-13b-chat-hf)
|
58 |
+
- Languages: Chinese/Japanese
|
59 |
+
|
60 |
+
### v0.5
|
61 |
+
|
62 |
+
- Finetuned by [SakuraUmi](https://github.com/pipixia244)
|
63 |
+
- Finetuned on [Baichuan2-13B-Chat](https://huggingface.co/baichuan-inc/Baichuan2-13B-Chat)
|
64 |
+
- Base model: [Baichuan2-13B-Base](https://huggingface.co/baichuan-inc/Baichuan2-13B-Base)
|
65 |
+
- Languages: Chinese/Japanese
|
66 |
+
|
67 |
+
### v0.7
|
68 |
+
|
69 |
+
- Finetuned by [SakuraUmi](https://github.com/pipixia244)
|
70 |
+
- Finetuned on [Qwen-14B-Chat](https://huggingface.co/Qwen/Qwen-14B)
|
71 |
+
- Base model: [Qwen-14B](https://huggingface.co/Qwen/Qwen-14B)
|
72 |
+
- Languages: Chinese/Japanese
|
73 |
+
|
74 |
+
## 版本
|
75 |
+
|
76 |
+
### v0.7
|
77 |
+
|
78 |
+
数据集:约1M Galgame中日文本 + 约3.4M 轻小说中日文本
|
79 |
+
|
80 |
+
微调方式:全参数
|
81 |
+
|
82 |
+
微调epoch数:2
|
83 |
+
|
84 |
+
参数量:14B
|
85 |
+
|
86 |
+
### v0.5
|
87 |
+
|
88 |
+
数据集:约260k Galgame中日文本 + 约2.3M 轻小说中日文本
|
89 |
+
|
90 |
+
微调方式:全参数
|
91 |
+
|
92 |
+
微调epoch数:2
|
93 |
+
|
94 |
+
参数量:13B
|
95 |
+
|
96 |
+
### v0.4
|
97 |
+
|
98 |
+
数据集:约260k Galgame中日文本 + 约2.3M 轻小说中日文本
|
99 |
+
|
100 |
+
微调方式:全参数
|
101 |
+
|
102 |
+
微调epoch数:1
|
103 |
+
|
104 |
+
参数量:13B
|
105 |
+
|
106 |
+
### v0.1
|
107 |
+
|
108 |
+
数据集:约260k Galgame中日文本
|
109 |
+
|
110 |
+
微调方式:全参数
|
111 |
+
|
112 |
+
微调epoch数:2
|
113 |
+
|
114 |
+
参数量:13B
|
115 |
+
|
116 |
+
## 效果
|
117 |
+
|
118 |
+
- Galgame
|
119 |
+
|
120 |
+
| 原文 | Ours(v0.5) | Ours(v0.4) | Ours(v0.1) | ChatGPT(GPT-3.5) |
|
121 |
+
| ---- | ---- | ---- | ---- | ---- |
|
122 |
+
| 「女の子の一人暮らしって、やっぱ一階は防范的に危ないのかな~? お父さんには、一階はやめとけ~って言われててね?」 | 「女生一个人住,一楼在防范上果然很危险吗~?我爸爸叫我不要住一楼哦。」 | 「女孩子一个人住,果然还是不太安全吧~?爸爸说过,不要住一楼~」 | 「一个女孩子住在一楼,还是有点不太安全吧?爸爸说让我不要住在一楼」 | "一个女孩子独自一人住,大概一楼会不安全吧~?爸爸对我说过,一楼最好不要住~" |
|
123 |
+
| 「助けて、誰か助けてって思いながら、ただただ泣いてたんです……」 | 「我一边想着谁来救救我,一边不停地哭……」 |「我一边想着有没有人能救救我,一边哭哭啼啼……」 | 「我一边祈祷着,祈祷着有人能来救救我们,一边不停地哭泣……」| 「帮帮我,我一边想着有人帮助我,一边只是哭泣着……」 |
|
124 |
+
| 「そうだよ。これが太一の普通の顔だって。でも、ちょっと不気味だから、わたしみたいにニッコリ笑ってみて?」 | 「对啊。这就是太一平常的表情。不过,这样有点毛骨悚然,所以试着像我这样笑吧?」 |「对啊,这就是太一的普通表情。不过,感觉有点诡异,你像我一样笑咪咪地试试看?」 |「是啊。这就是太一的普通表情。但是,因为有点吓人,所以你也试着像我一样微笑一下吧?」 | “是的呢,这就是太一的平常表情哦。不过,有点怪异,所以像我这样放个甜甜的笑容试试看?” |
|
125 |
+
| 「そういうヒトの感情は、発情期を迎えてもいないネコには難しい」 | 「对于还没到发情期的猫来说,人类的这种感情实在是有点难以理解」| 「这种人类的感情,对还没进入发情期的猫来说太难懂了。」 |「这种人类的感情,对还没有迎来发情期的猫来说太难懂了」 | 这种人类的情感对于尚未进入发情期的猫来说是复杂的。 |
|
126 |
+
| 「朝になって、病院に行くまで。ずっと、ずーっとそばに居てくれて……」 | 「从早上到去医院的这段时间,一直、一直陪在我身边……」 | 「一直陪伴着我,直到早上去医院为止……」 |「一直陪我到早上去医院。一直,一直陪在我身边……」 | "直到早晨去医院为止。一直,一直都在我身旁……" |
|
127 |
+
| 「それ以外は、自由に過ごしているため、各自が好きにできる、とても平和な部活だった……。」 | 「除此之外,由于可以自由活动,大家都能随心所欲,所以是非常和平的社团活动……」 |「除此之外,我们都可以自由活动,每个人都能随心所欲,是个非常和平的社团……」 | 「除此之外,社团活动都是自由参加的,每个人都可以按自己的意愿去做自己想做的事情,所以社团活动也是非常和平的……」 | 「除此以外,因为大家都自由自在地度过时间,是个每个人都能按自己喜好随意参与的非常和平的社团活动......。」|
|
128 |
+
| 「そーだそーだ。せっかくお店休みにして遊びに来たのに」 | 「对呀对呀,难得我们关店跑出来玩耶。」 | 「没错没错,难得店里放假,我们才来玩的。」 |「是啊是啊,难得休息一天,我还想出来玩一下呢」| "是啊是啊。本来店铺难得休息,特地过来玩的呢。" |
|
129 |
+
| 伝えなければ、伝わらない。きっと、大事なことであるほど。 | 不表达就传达不了。越是重要的事情,越是如此。 | 不说出来就不会知道。越是重要的事情,就越是不能不说。 | -- | 如果不传达,就不会传达。毫无疑问,对于重要的事情来说是如此。 |
|
130 |
+
| が、ハチロクを手伝うことでそれが果たせるというのなら、仕事がどれほど増えようと、決して苦とは感じない。 | 不过,如果帮忙八六能让我实现这个愿望,无论工作增加多少,我也绝不会觉得痛苦。 | 不过,如果帮助八六就能实现这个愿望,不管工作多么繁重,我都不会觉得辛苦。 | -- | 如果通过帮助八六实现这一目标,无论工作增加多少,我绝不会感到苦恼。 |
|
131 |
+
|
132 |
+
- 轻小说(全龄)
|
133 |
+
|
134 |
+
使用[该仓库](https://github.com/FishHawk/sakura-test)的测试文本,仓库内提供了测试代码,测试文本以及v0.5版本的测试结果。使用该仓库代码在v0.7模型上对测试文本进行测试的结果已上传到[sakuraumi/Sakura-13B-Galgame-Archived](https://huggingface.co/sakuraumi/Sakura-13B-Galgame-Archived/blob/main/text.sa-packed)。
|
135 |
+
|
136 |
+
- 轻小说(R18)
|
137 |
+
|
138 |
+
使用[这本r18小说](https://novel18.syosetu.com/n2975ig)的部分文本,比较各版本输出文本(其中GPT网页端和API均无法翻译),结果已上传至[链接](https://huggingface.co/sakuraumi/Sakura-13B-Galgame-Archived/blob/main/r18_test.txt)
|
139 |
+
|
140 |
+
# 推理
|
141 |
+
|
142 |
+
- Galgame翻译的prompt构建:
|
143 |
+
|
144 |
+
- v0.1
|
145 |
+
|
146 |
+
```python
|
147 |
+
input_text = "" # 用户输入
|
148 |
+
query = "将下面的日文文本翻译成中文:" + input_text
|
149 |
+
prompt = "Human: \n" + query + "\n\nAssistant: \n"
|
150 |
+
```
|
151 |
+
|
152 |
+
- v0.4
|
153 |
+
|
154 |
+
```python
|
155 |
+
input_text = "" # 用户输入
|
156 |
+
query = "将下面的日文文本翻译成中文:" + input_text
|
157 |
+
prompt = "User: " + query + "\nAssistant: "
|
158 |
+
```
|
159 |
+
|
160 |
+
- v0.5+
|
161 |
+
|
162 |
+
```python
|
163 |
+
input_text = "" # 用户输入
|
164 |
+
query = "将下面的日文文本翻译成中文:" + input_text
|
165 |
+
prompt = "<reserved_106>" + query + "<reserved_107>"
|
166 |
+
```
|
167 |
+
|
168 |
+
- v0.7
|
169 |
+
参考Qwen-14B-Chat的prompt构造方式:[这里](https://huggingface.co/Qwen/Qwen-14B-Chat/blob/5188dfeb4ff175705aa3a84ef9d616c70dea029b/qwen_generation_utils.py#L119)和[这里](https://github.com/hiyouga/LLaMA-Efficient-Tuning/blob/5310e4d1829f36619c8f224d09ec15eeaf7a4877/src/llmtuner/extras/template.py#L546)
|
170 |
+
|
171 |
+
|
172 |
+
- 推理与解码参数:
|
173 |
+
|
174 |
+
| 参数 | 值 |
|
175 |
+
| ---- | ---- |
|
176 |
+
| temperature | 1 |
|
177 |
+
| top p | 0.5 |
|
178 |
+
| do sample | True |
|
179 |
+
| beams number | 1 |
|
180 |
+
| repetition penalty | 1 |
|
181 |
+
| max new token | 512 |
|
182 |
+
| min new token | 1 |
|
183 |
+
|
184 |
+
- 量化:
|
185 |
+
|
186 |
+
根据transformers文档中给出的AutoGPTQ量化教程自行量化,或使用我们已经量化好的模型。
|
187 |
+
|
188 |
+
# 微调
|
189 |
+
|
190 |
+
流程与LLaMA2(v0.1-v0.4)/Baichuan2(v0.5+)/Qwen14B(v0.7)一致,prompt构造参考推理部分
|
191 |
+
|
192 |
+
# 后续工作
|
193 |
+
|
194 |
+
1. 优化数据集,主要优化数据集质量
|
195 |
+
2. 支持上下文理解
|
196 |
+
3. 支持指定专有名词
|
197 |
+
|
198 |
+
# 致谢
|
199 |
+
|
200 |
+
- [CjangCjengh](https://github.com/CjangCjengh)
|
201 |
+
|
202 |
+
- [ryank231231](https://github.com/ryank231231)
|
203 |
+
|
204 |
+
- 三日月クリ
|
205 |
+
|
206 |
+
- [K024](https://github.com/K024)
|
207 |
+
|
208 |
+
- [minaduki-sora](https://github.com/minaduki-sora)
|
209 |
+
|
210 |
+
- [Kimagure7](https://github.com/Kimagure7)
|
211 |
+
|
212 |
+
- [YYF233333](https://github.com/YYF233333)
|
213 |
+
|
214 |
+
# Copyright Notice
|
215 |
+
|
216 |
+
This model is built upon Meta's LLaMA series of models and is subject to Meta's licensing agreement.
|
217 |
+
|
218 |
+
This model is intended for use only by individuals who have obtained approval from Meta and are eligible to download LLaMA.
|
219 |
+
|
220 |
+
If you have not obtained approval from Meta, you must visit the https://ai.meta.com/llama/ page, read and agree to the model's licensing agreement, submit an application, and wait for approval from Meta before downloading the model from this page.
|
221 |
+
|