models.py

METHOD_OPTIONS = {
    "nli": "Natural Language Inference",
    "nsp": "Next Sentence Prediction (not supported yet)",
}

NLI_MODEL_OPTIONS = [
    "emrecan/distilbert-base-turkish-cased-allnli_tr",
    "emrecan/distilbert-base-turkish-cased-multinli_tr",
    "emrecan/distilbert-base-turkish-cased-snli_tr",
    "emrecan/bert-base-turkish-cased-allnli_tr",
    "emrecan/bert-base-turkish-cased-multinli_tr",
    "emrecan/bert-base-turkish-cased-snli_tr",
    "emrecan/convbert-base-turkish-mc4-cased-allnli_tr",
    "emrecan/convbert-base-turkish-mc4-cased-multinli_tr",
    "emrecan/convbert-base-turkish-mc4-cased-snli_tr",
    "emrecan/bert-base-multilingual-cased-allnli_tr",
    "emrecan/bert-base-multilingual-cased-multinli_tr",
    "emrecan/bert-base-multilingual-cased-snli_tr",
]

NSP_MODEL_OPTIONS = [
    "dbmdz/bert-base-turkish-cased",
    "dbmdz/bert-base-turkish-uncased",
    "dbmdz/bert-base-turkish-128k-cased",
    "dbmdz/bert-base-turkish-128k-uncased",
]