dmedhi commited on
Commit
fe68b4f
·
verified ·
1 Parent(s): 558b455

Fix config for sentence-transformers compatibility

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "PawanEmbdModel"
4
  ],
5
  "bos_token_id": 0,
6
  "dropout": 0.1,
@@ -9,7 +9,7 @@
9
  "hidden_size": 256,
10
  "intermediate_size": 1024,
11
  "max_position_embeddings": 512,
12
- "model_type": "pawan_embd",
13
  "num_heads": 4,
14
  "num_layers": 4,
15
  "output_size": 768,
 
1
  {
2
  "architectures": [
3
+ "XLMRobertaModel"
4
  ],
5
  "bos_token_id": 0,
6
  "dropout": 0.1,
 
9
  "hidden_size": 256,
10
  "intermediate_size": 1024,
11
  "max_position_embeddings": 512,
12
+ "model_type": "xlm-roberta",
13
  "num_heads": 4,
14
  "num_layers": 4,
15
  "output_size": 768,