Skip to content

fix(medcat-plugins - transformer-NER) Changed to AutoModel from AutoModelForTokenClassification - #645

Open
adam-sutton-1992 wants to merge 12 commits into
mainfrom
transformer_ner_AutoModel_fix
Open

adam-sutton-1992 wants to merge 12 commits into
mainfrom
transformer_ner_AutoModel_fix

Conversation

@adam-sutton-1992

Copy link
Copy Markdown
Contributor

Just a minor fix. Some language models don't have TokenClassifications defined, but will always have a modelling body.

I've added a classifier head also for the CRF emissions.

@mart-r

mart-r commented Oct 23, 2026

Copy link
Copy Markdown
Collaborator

This isn't in sync with main so might present changes from other PRs as well (i.e from #639 which I just merged). So might want to resync (i.e merge main into this).

@mart-r mart-r left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice and clean - fully transfromers-ner - change now.

Looks good overall.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants