minhtoan commited on
Commit
e1ae555
·
1 Parent(s): 51287b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -4,6 +4,7 @@ tags:
4
  - vi
5
  - vietnamese
6
  - gpt2
 
7
  - text-generation
8
  - lm
9
  - nlp
@@ -14,12 +15,12 @@ widget:
14
 
15
  inference:
16
  parameters:
17
- max_length: 80
18
  ---
19
 
20
- # GPT-2
21
 
22
- Pretrained gpt3 small model (gpt-neo) on Vietnamese for text generation
23
 
24
  # How to use the model
25
 
@@ -31,7 +32,7 @@ model = GPTNeoForCausalLM.from_pretrained('minhtoan/gpt3-small-vietnamese')
31
 
32
  text = "Hoa quả và rau thường rẻ hơn khi vào mùa"
33
  input_ids = tokenizer.encode(text, return_tensors='pt')
34
- max_length = 80
35
 
36
  sample_outputs = model.generate(input_ids, do_sample=True, max_length=max_length)
37
 
 
4
  - vi
5
  - vietnamese
6
  - gpt2
7
+ - gpt3
8
  - text-generation
9
  - lm
10
  - nlp
 
15
 
16
  inference:
17
  parameters:
18
+ max_length: 100
19
  ---
20
 
21
+ # GPT-3 small
22
 
23
+ Pretrained GPT Neo, it's architecture intentionally resembles that of GPT-3, model was trained on Vietnamese dataset for text generation
24
 
25
  # How to use the model
26
 
 
32
 
33
  text = "Hoa quả và rau thường rẻ hơn khi vào mùa"
34
  input_ids = tokenizer.encode(text, return_tensors='pt')
35
+ max_length = 100
36
 
37
  sample_outputs = model.generate(input_ids, do_sample=True, max_length=max_length)
38