universal-dependencies/universal_dependencies
Viewer • Updated • 2.22M • 7.6k • 6
How to use ChanBeDu/vivtb_electra-based_model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="ChanBeDu/vivtb_electra-based_model") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("ChanBeDu/vivtb_electra-based_model")
model = AutoModelForTokenClassification.from_pretrained("ChanBeDu/vivtb_electra-based_model", device_map="auto")Overview: This model is finetuned based on NlpHUST vietnamese electra based model. The used datased is Universial dependencies dataset. Only the tree branch vi_vtb of the dataset is used for finetuning. The model is used for UPOS tagging task