zero-shot-turkish / models.py
emrecan's picture
first working app
ddcecdf
raw
history blame
No virus
958 Bytes
METHOD_OPTIONS = {
"nli": "Natural Language Inference",
"nsp": "Next Sentence Prediction",
}
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",
]