How do we customize the number of entites for NER pretrained model?

🤗Tokenizers
1

Hi,
I am trying to re-train the model with custom dataset with X entites. I found the documentation link: AutoModels — transformers 3.0.2 documentation
AutoModelForTokenClassification.from_pretrained(“bert-base-multilingual-cased”)

Anyone knows what is the configuration to set the pre-trained huggingface model to X entities?

Thank you,
Dinesh

2

AutoModelForTokenClassification.from_pretrained(“bert-base-multilingual-cased”, num_labels=X)