XiweiZ commited on
Commit
6362b71
1 Parent(s): 35b5656

Add SetFit model

Browse files
1_Pooling/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "word_embedding_dimension": 768,
3
- "pooling_mode_cls_token": true,
4
- "pooling_mode_mean_tokens": false,
5
  "pooling_mode_max_tokens": false,
6
  "pooling_mode_mean_sqrt_len_tokens": false
7
  }
 
1
  {
2
  "word_embedding_dimension": 768,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
  "pooling_mode_max_tokens": false,
6
  "pooling_mode_mean_sqrt_len_tokens": false
7
  }
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/datascience/.cache/torch/sentence_transformers/sentence-transformers_multi-qa-mpnet-base-dot-v1/",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "/home/datascience/.cache/torch/sentence_transformers/sentence-transformers_paraphrase-mpnet-base-v2/",
3
  "architectures": [
4
  "MPNetModel"
5
  ],
config_sentence_transformers.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "__version__": {
3
  "sentence_transformers": "2.0.0",
4
- "transformers": "4.6.1",
5
- "pytorch": "1.8.1"
6
  }
7
  }
 
1
  {
2
  "__version__": {
3
  "sentence_transformers": "2.0.0",
4
+ "transformers": "4.7.0",
5
+ "pytorch": "1.9.0+cu102"
6
  }
7
  }
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:13fe61d92cef49c5324dad0ad1f759a563d2421fb2399b1649eef6ef581e4d41
3
  size 6991
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84998b8483e4df7ab09aed7496b032cc922f079bc47fc6d29d2a7c3047a05e5b
3
  size 6991
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:14c1c24e22cd448f15a58d5bbe5747010860668028f20cdfdda3c4006e01371e
3
  size 438016493
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45496c68bd7c7cfc85678bdaeeb9c6676f6420b2629875e9db0ddea0b8101fac
3
  size 438016493
tokenizer.json CHANGED
@@ -42,15 +42,6 @@
42
  "normalized": false,
43
  "special": true
44
  },
45
- {
46
- "id": 3,
47
- "content": "<unk>",
48
- "single_word": false,
49
- "lstrip": false,
50
- "rstrip": false,
51
- "normalized": true,
52
- "special": true
53
- },
54
  {
55
  "id": 104,
56
  "content": "[UNK]",
@@ -81,17 +72,85 @@
81
  "type": "BertPreTokenizer"
82
  },
83
  "post_processor": {
84
- "type": "RobertaProcessing",
85
- "sep": [
86
- "</s>",
87
- 2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
88
  ],
89
- "cls": [
90
- "<s>",
91
- 0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
92
  ],
93
- "trim_offsets": true,
94
- "add_prefix_space": false
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
95
  },
96
  "decoder": {
97
  "type": "WordPiece",
 
42
  "normalized": false,
43
  "special": true
44
  },
 
 
 
 
 
 
 
 
 
45
  {
46
  "id": 104,
47
  "content": "[UNK]",
 
72
  "type": "BertPreTokenizer"
73
  },
74
  "post_processor": {
75
+ "type": "TemplateProcessing",
76
+ "single": [
77
+ {
78
+ "SpecialToken": {
79
+ "id": "<s>",
80
+ "type_id": 0
81
+ }
82
+ },
83
+ {
84
+ "Sequence": {
85
+ "id": "A",
86
+ "type_id": 0
87
+ }
88
+ },
89
+ {
90
+ "SpecialToken": {
91
+ "id": "</s>",
92
+ "type_id": 0
93
+ }
94
+ }
95
  ],
96
+ "pair": [
97
+ {
98
+ "SpecialToken": {
99
+ "id": "<s>",
100
+ "type_id": 0
101
+ }
102
+ },
103
+ {
104
+ "Sequence": {
105
+ "id": "A",
106
+ "type_id": 0
107
+ }
108
+ },
109
+ {
110
+ "SpecialToken": {
111
+ "id": "</s>",
112
+ "type_id": 0
113
+ }
114
+ },
115
+ {
116
+ "SpecialToken": {
117
+ "id": "</s>",
118
+ "type_id": 0
119
+ }
120
+ },
121
+ {
122
+ "Sequence": {
123
+ "id": "B",
124
+ "type_id": 1
125
+ }
126
+ },
127
+ {
128
+ "SpecialToken": {
129
+ "id": "</s>",
130
+ "type_id": 1
131
+ }
132
+ }
133
  ],
134
+ "special_tokens": {
135
+ "</s>": {
136
+ "id": "</s>",
137
+ "ids": [
138
+ 2
139
+ ],
140
+ "tokens": [
141
+ "</s>"
142
+ ]
143
+ },
144
+ "<s>": {
145
+ "id": "<s>",
146
+ "ids": [
147
+ 0
148
+ ],
149
+ "tokens": [
150
+ "<s>"
151
+ ]
152
+ }
153
+ }
154
  },
155
  "decoder": {
156
  "type": "WordPiece",
tokenizer_config.json CHANGED
@@ -1,15 +1,66 @@
1
  {
2
- "bos_token": "<s>",
 
 
 
 
 
 
 
3
  "clean_up_tokenization_spaces": true,
4
- "cls_token": "<s>",
 
 
 
 
 
 
 
 
5
  "do_lower_case": true,
6
- "eos_token": "</s>",
7
- "mask_token": "<mask>",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  "model_max_length": 512,
9
- "pad_token": "<pad>",
10
- "sep_token": "</s>",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  "strip_accents": null,
12
  "tokenize_chinese_chars": true,
13
  "tokenizer_class": "MPNetTokenizer",
14
- "unk_token": "[UNK]"
 
 
 
 
 
 
 
15
  }
 
1
  {
2
+ "bos_token": {
3
+ "__type": "AddedToken",
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": true,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
  "clean_up_tokenization_spaces": true,
11
+ "cls_token": {
12
+ "__type": "AddedToken",
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "do_basic_tokenize": true,
20
  "do_lower_case": true,
21
+ "eos_token": {
22
+ "__type": "AddedToken",
23
+ "content": "</s>",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false
28
+ },
29
+ "mask_token": {
30
+ "__type": "AddedToken",
31
+ "content": "<mask>",
32
+ "lstrip": true,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ },
37
  "model_max_length": 512,
38
+ "never_split": null,
39
+ "pad_token": {
40
+ "__type": "AddedToken",
41
+ "content": "<pad>",
42
+ "lstrip": false,
43
+ "normalized": true,
44
+ "rstrip": false,
45
+ "single_word": false
46
+ },
47
+ "sep_token": {
48
+ "__type": "AddedToken",
49
+ "content": "</s>",
50
+ "lstrip": false,
51
+ "normalized": true,
52
+ "rstrip": false,
53
+ "single_word": false
54
+ },
55
  "strip_accents": null,
56
  "tokenize_chinese_chars": true,
57
  "tokenizer_class": "MPNetTokenizer",
58
+ "unk_token": {
59
+ "__type": "AddedToken",
60
+ "content": "[UNK]",
61
+ "lstrip": false,
62
+ "normalized": true,
63
+ "rstrip": false,
64
+ "single_word": false
65
+ }
66
  }