Extending AutoModel class from Hugging Face
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.