SkyWork
commited on
Commit
•
c72ad27
1
Parent(s):
326f068
Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
#
|
2 |
|
3 |
-
|
4 |
|
5 |
|
6 |
## 项目亮点
|
@@ -36,8 +36,8 @@ from transformers import GPT2LMHeadModel
|
|
36 |
from transformers import AutoTokenizer
|
37 |
from transformers import TextGenerationPipeline
|
38 |
|
39 |
-
model = GPT2LMHeadModel.from_pretrained("SkyWork/
|
40 |
-
tokenizer = AutoTokenizer.from_pretrained("SkyWork/
|
41 |
text_generator = TextGenerationPipeline(model, tokenizer, device=0)
|
42 |
input_str = "今天是个好天气"
|
43 |
max_new_tokens = 20
|
|
|
1 |
+
# SkyTextTiny
|
2 |
|
3 |
+
SkyTextTiny是由奇点智源发布的中文GPT3预训练模型,参数量30亿左右,可以进行聊天、问答、中英互译等不同的[任务](https://openapi.singularity-ai.com/index.html#/examplesIndex)。
|
4 |
|
5 |
|
6 |
## 项目亮点
|
|
|
36 |
from transformers import AutoTokenizer
|
37 |
from transformers import TextGenerationPipeline
|
38 |
|
39 |
+
model = GPT2LMHeadModel.from_pretrained("SkyWork/SkyTextTiny")
|
40 |
+
tokenizer = AutoTokenizer.from_pretrained("SkyWork/SkyTextTiny", trust_remote_code=True)
|
41 |
text_generator = TextGenerationPipeline(model, tokenizer, device=0)
|
42 |
input_str = "今天是个好天气"
|
43 |
max_new_tokens = 20
|