Extending AutoModel class from Hugging Face

Anuj Arora
ML@ABEJA
Published in
2 min readApr 2, 2023

--

A hacky way to keep the code structure consistent

In this article I share the method for customising the model architecture of Hugging Face’s AutoModel series of models.

Reference Task: AutoModelForTokenClassification (ner)*
Library Version: transformers: 4.27.4

Possible Solution

Through extensive google search, I found a plausible solution.

--

--