Update README.md
Browse files
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:
|
18 |
---
|
19 |
|
20 |
-
# GPT-
|
21 |
|
22 |
-
Pretrained
|
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 =
|
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 |
|