saintway commited on
Commit
b2c21aa
1 Parent(s): ddfba58

utilizing sentence_transformers

Browse files

```python
from sentence_transformers import SentenceTransformer
SentenceTransformer('shibing624/text2vec-base-chinese')
```

Files changed (3) hide show
  1. config.json +2 -30
  2. modules.json +14 -0
  3. sentence_bert_config.json +4 -0
config.json CHANGED
@@ -1,32 +1,4 @@
1
  {
2
- "_name_or_path": "hfl/chinese-macbert-base",
3
- "architectures": [
4
- "BertModel"
5
- ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "classifier_dropout": null,
8
- "directionality": "bidi",
9
- "gradient_checkpointing": false,
10
- "hidden_act": "gelu",
11
- "hidden_dropout_prob": 0.1,
12
- "hidden_size": 768,
13
- "initializer_range": 0.02,
14
- "intermediate_size": 3072,
15
- "layer_norm_eps": 1e-12,
16
- "max_position_embeddings": 512,
17
- "model_type": "bert",
18
- "num_attention_heads": 12,
19
- "num_hidden_layers": 12,
20
- "pad_token_id": 0,
21
- "pooler_fc_size": 768,
22
- "pooler_num_attention_heads": 12,
23
- "pooler_num_fc_layers": 3,
24
- "pooler_size_per_head": 128,
25
- "pooler_type": "first_token_transform",
26
- "position_embedding_type": "absolute",
27
- "torch_dtype": "float32",
28
- "transformers_version": "4.12.3",
29
- "type_vocab_size": 2,
30
- "use_cache": true,
31
- "vocab_size": 21128
32
  }
 
1
  {
2
+ "word_embedding_dimension": 768,
3
+ "pooling_mode_mean_tokens": true
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  }
modules.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.models.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.models.Pooling"
13
+ }
14
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "max_seq_length": 128,
3
+ "do_lower_case": false
4
+ }