```
from transformers import AutoModelForCausalLM

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