add model
Browse files- config.json +6 -2
- flax_model.msgpack +3 -0
config.json
CHANGED
|
@@ -27,6 +27,8 @@
|
|
| 27 |
"eos_token_id": 2,
|
| 28 |
"extra_pos_embeddings": 2,
|
| 29 |
"force_bos_token_to_be_generated": false,
|
|
|
|
|
|
|
| 30 |
"id2label": {
|
| 31 |
"0": "HALLUCINATED",
|
| 32 |
"1": "FAITHFUL"
|
|
@@ -34,8 +36,8 @@
|
|
| 34 |
"init_std": 0.02,
|
| 35 |
"is_encoder_decoder": true,
|
| 36 |
"label2id": {
|
| 37 |
-
"
|
| 38 |
-
"
|
| 39 |
},
|
| 40 |
"max_position_embeddings": 1024,
|
| 41 |
"model_type": "bart",
|
|
@@ -67,5 +69,7 @@
|
|
| 67 |
"num_beams": 6
|
| 68 |
}
|
| 69 |
},
|
|
|
|
|
|
|
| 70 |
"vocab_size": 50265
|
| 71 |
}
|
|
|
|
| 27 |
"eos_token_id": 2,
|
| 28 |
"extra_pos_embeddings": 2,
|
| 29 |
"force_bos_token_to_be_generated": false,
|
| 30 |
+
"forced_eos_token_id": 2,
|
| 31 |
+
"gradient_checkpointing": false,
|
| 32 |
"id2label": {
|
| 33 |
"0": "HALLUCINATED",
|
| 34 |
"1": "FAITHFUL"
|
|
|
|
| 36 |
"init_std": 0.02,
|
| 37 |
"is_encoder_decoder": true,
|
| 38 |
"label2id": {
|
| 39 |
+
"FAITHFUL": 1,
|
| 40 |
+
"HALLUCINATED": 0
|
| 41 |
},
|
| 42 |
"max_position_embeddings": 1024,
|
| 43 |
"model_type": "bart",
|
|
|
|
| 69 |
"num_beams": 6
|
| 70 |
}
|
| 71 |
},
|
| 72 |
+
"transformers_version": "4.7.0.dev0",
|
| 73 |
+
"use_cache": true,
|
| 74 |
"vocab_size": 50265
|
| 75 |
}
|
flax_model.msgpack
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fdc5abb901a360d7f6bbd39925c3db787a3f88d4580531d02de2260defd155fa
|
| 3 |
+
size 560059477
|