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
|