facebook/bart-large-cnn
facebook/bart-large-cnn
1
0
Overview Code Discussions Pull Requests Wiki Activity

add flax model

Browse Source
This commit is contained in:
patil-suraj 2021-06-14 07:28:15 +00:00
parent 18614750f2
commit 7676c0925b
3 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitattributes vendored
View File

@ -6,3 +6,4 @@
*.tar.gz filter=lfs diff=lfs merge=lfs -text *.tar.gz filter=lfs diff=lfs merge=lfs -text
*.ot filter=lfs diff=lfs merge=lfs -text *.ot filter=lfs diff=lfs merge=lfs -text
*.onnx filter=lfs diff=lfs merge=lfs -text *.onnx filter=lfs diff=lfs merge=lfs -text
*.msgpack filter=lfs diff=lfs merge=lfs -text

7
config.json
View File

@ -3,9 +3,13 @@
"activation_dropout": 0.0, "activation_dropout": 0.0,
"activation_function": "gelu", "activation_function": "gelu",
"add_final_layer_norm": false, "add_final_layer_norm": false,
"architectures": [
"BartForConditionalGeneration"
],
"attention_dropout": 0.0, "attention_dropout": 0.0,
"bos_token_id": 0, "bos_token_id": 0,
"classif_dropout": 0.0, "classif_dropout": 0.0,
"classifier_dropout": 0.0,
"d_model": 1024, "d_model": 1024,
"decoder_attention_heads": 16, "decoder_attention_heads": 16,
"decoder_ffn_dim": 4096, "decoder_ffn_dim": 4096,
@ -22,6 +26,7 @@
"force_bos_token_to_be_generated": true, "force_bos_token_to_be_generated": true,
"forced_bos_token_id": 0, "forced_bos_token_id": 0,
"forced_eos_token_id": 2, "forced_eos_token_id": 2,
"gradient_checkpointing": false,
"id2label": { "id2label": {
"0": "LABEL_0", "0": "LABEL_0",
"1": "LABEL_1", "1": "LABEL_1",
@ -57,5 +62,7 @@
"num_beams": 4 "num_beams": 4
} }
}, },
"transformers_version": "4.7.0.dev0",
"use_cache": true,
"vocab_size": 50264 "vocab_size": 50264
} }

BIN
flax_model.msgpack (Stored with Git LFS) Normal file
View File

Binary file not shown.