from transformers import AutoModel, AutoTokenizer


tokenizer = AutoTokenizer.from_pretrained('truongpdd/vi-en-roberta-base')
model = AutoModel.from_pretrained('truongpdd/vi-en-roberta-base', from_flax=True)