| from transformers import AutoModelForTokenClassification, AutoModelForQuestionAnswering | |
| from src.utils.mapper import configmapper | |
| configmapper.map("models", "autotoken")(AutoModelForTokenClassification) | |
| configmapper.map("models", "autotoken_3cls")(AutoModelForTokenClassification) | |
| configmapper.map("models", "autospans")(AutoModelForQuestionAnswering) | |