facebook
/
blenderbot-400M-distill
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add flax model

This commit is contained in:
patil-suraj 2021-11-30 12:15:51 +01:00
parent 2ebb89337d
commit f672f129aa
3 changed files with 7 additions and 1 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

4
config.json
View File

@ -23,10 +23,12 @@
"encoder_ffn_dim": 5120, "encoder_ffn_dim": 5120,
"encoder_layerdrop": 0.0, "encoder_layerdrop": 0.0,
"encoder_layers": 2, "encoder_layers": 2,
"encoder_no_repeat_ngram_size": 3,
"eos_token_id": 2, "eos_token_id": 2,
"extra_layer_norm": false, "extra_layer_norm": false,
"extra_pos_embeddings": 0, "extra_pos_embeddings": 0,
"force_bos_token_to_be_generated": false, "force_bos_token_to_be_generated": false,
"forced_eos_token_id": 2,
"gradient_checkpointing": false, "gradient_checkpointing": false,
"id2label": { "id2label": {
"0": "LABEL_0", "0": "LABEL_0",
@ -54,7 +56,7 @@
"pad_token_id": 0, "pad_token_id": 0,
"scale_embedding": true, "scale_embedding": true,
"static_position_embeddings": false, "static_position_embeddings": false,
"transformers_version": "4.2.0dev0", "transformers_version": "4.13.0.dev0",
"unk_token_id": 3, "unk_token_id": 3,
"use_cache": true, "use_cache": true,
"vocab_size": 8008 "vocab_size": 8008

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

Binary file not shown.