File size: 353 Bytes
cc9c7ee |
1 2 3 4 5 6 7 |
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)
|