vietnews-gpt2 / README.md
truongpdd's picture
Update README.md
bd422f1
|
raw
history blame
274 Bytes

How to use:

from transformers import AutoModelForCausalLM, AutoTokenizer


tokenizer = AutoTokenizer.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs')
model = AutoModelForCausalLM.from_pretrained('truongpdd/vietnamese-news-gpt2-5epochs', from_flax=True)