Hugging Face's logo

truongpdd
/
vietnews-gpt2

Text Generation
Transformers
PyTorch
JAX
TensorBoard
gpt2
text-generation-inference
Inference Endpoints
Model card Files Metrics Community
1
truongpdd commited on
Commit
bd422f1
•
1 Parent(s): cbe4e90

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -1,7 +1,9 @@
1
-
2
 
3
  ```
4
- from transformers import AutoModelForCausalLM
 
5
 
 
6
  model = AutoModelForCausalLM.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs', from_flax=True)
7
  ```
 
1
+ ## How to use:
2
 
3
  ```
4
+ from transformers import AutoModelForCausalLM, AutoTokenizer
5
+
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs')
8
  model = AutoModelForCausalLM.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs', from_flax=True)
9
  ```