SkyWork commited on
Commit
c72ad27
1 Parent(s): 326f068

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -1,6 +1,6 @@
1
- # SkyTextJunior
2
 
3
- SkyTextJunior是由奇点智源发布的中文GPT3预训练模型,参数量30亿左右,可以进行聊天、问答、中英互译等不同的[任务](https://openapi.singularity-ai.com/index.html#/examplesIndex)。
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/SkyTextJunior")
40
- tokenizer = AutoTokenizer.from_pretrained("SkyWork/SkyTextJunior", trust_remote_code=True)
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