fschlatt commited on
Commit
7ba514f
1 Parent(s): 54b747c

update to new lightning-ir version

Browse files
Files changed (2) hide show
  1. config.json +7 -4
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,8 +1,10 @@
1
  {
 
2
  "architectures": [
3
- "FlashMonoElectraModel"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
6
  "classifier_dropout": null,
7
  "doc_length": 256,
8
  "embedding_size": 768,
@@ -13,10 +15,11 @@
13
  "intermediate_size": 3072,
14
  "layer_norm_eps": 1e-12,
15
  "max_position_embeddings": 512,
16
- "model_type": "mono-electra",
17
  "num_attention_heads": 12,
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 0,
 
20
  "position_embedding_type": "absolute",
21
  "query_length": 32,
22
  "summary_activation": "gelu",
@@ -24,8 +27,8 @@
24
  "summary_type": "first",
25
  "summary_use_proj": true,
26
  "torch_dtype": "float32",
27
- "transformers_version": "4.39.3",
28
  "type_vocab_size": 2,
29
  "use_cache": true,
30
  "vocab_size": 30522
31
- }
 
1
  {
2
+ "_name_or_path": "/home/fschlatt/Models/monoelectra-base",
3
  "architectures": [
4
+ "CrossEncoderElectraModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
+ "backbone_model_type": "electra",
8
  "classifier_dropout": null,
9
  "doc_length": 256,
10
  "embedding_size": 768,
 
15
  "intermediate_size": 3072,
16
  "layer_norm_eps": 1e-12,
17
  "max_position_embeddings": 512,
18
+ "model_type": "cross-encoder",
19
  "num_attention_heads": 12,
20
  "num_hidden_layers": 12,
21
  "pad_token_id": 0,
22
+ "pooling_strategy": "first",
23
  "position_embedding_type": "absolute",
24
  "query_length": 32,
25
  "summary_activation": "gelu",
 
27
  "summary_type": "first",
28
  "summary_use_proj": true,
29
  "torch_dtype": "float32",
30
+ "transformers_version": "4.41.2",
31
  "type_vocab_size": 2,
32
  "use_cache": true,
33
  "vocab_size": 30522
34
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:80f97b4916513eb72c0d73865b383e6b58ae9378720837568cfde84db96ccd3f
3
- size 435593596
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:763a09add59ddfc02d87d3211d40374769b2c7060f92507a885f3917a33f7caa
3
+ size 435592020