emrecan commited on
Commit
7e9d867
1 Parent(s): 291ac4c

add options

Browse files
Files changed (1) hide show
  1. models.py +26 -0
models.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ METHOD_OPTIONS = {
2
+ "nli": "Natural Language Inference",
3
+ "nsp": "Next Sentence Prediction (not supported yet)",
4
+ }
5
+
6
+ NLI_MODEL_OPTIONS = [
7
+ "emrecan/distilbert-base-turkish-cased-allnli_tr",
8
+ "emrecan/distilbert-base-turkish-cased-multinli_tr",
9
+ "emrecan/distilbert-base-turkish-cased-snli_tr",
10
+ "emrecan/bert-base-turkish-cased-allnli_tr",
11
+ "emrecan/bert-base-turkish-cased-multinli_tr",
12
+ "emrecan/bert-base-turkish-cased-snli_tr",
13
+ "emrecan/convbert-base-turkish-mc4-cased-allnli_tr",
14
+ "emrecan/convbert-base-turkish-mc4-cased-multinli_tr",
15
+ "emrecan/convbert-base-turkish-mc4-cased-snli_tr",
16
+ "emrecan/bert-base-multilingual-cased-allnli_tr",
17
+ "emrecan/bert-base-multilingual-cased-multinli_tr",
18
+ "emrecan/bert-base-multilingual-cased-snli_tr",
19
+ ]
20
+
21
+ NSP_MODEL_OPTIONS = [
22
+ "dbmdz/bert-base-turkish-cased",
23
+ "dbmdz/bert-base-turkish-uncased",
24
+ "dbmdz/bert-base-turkish-128k-cased",
25
+ "dbmdz/bert-base-turkish-128k-uncased",
26
+ ]