# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-vietnamese")
model = AutoModelForMaskedLM.from_pretrained("KoichiYasuoka/roberta-base-vietnamese", device_map="auto")Quick Links
roberta-base-vietnamese
Model Description
This is a RoBERTa model pre-trained on Vietnamese Wikipedia texts. NVIDIA A100-SXM4-40GB took 20 hours 11 minutes for training. You can fine-tune roberta-base-vietnamese for downstream tasks, such as POS-tagging, dependency-parsing, and so on.
How to Use
from transformers import AutoTokenizer,AutoModelForMaskedLM
tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/roberta-base-vietnamese")
model=AutoModelForMaskedLM.from_pretrained("KoichiYasuoka/roberta-base-vietnamese")
- Downloads last month
- 74
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="KoichiYasuoka/roberta-base-vietnamese")