Add tf_weights

This commit is contained in:
Arthur Zucker 2022-06-03 12:17:16 +00:00
parent bf21b09d70
commit cff2d3dae8
2 changed files with 8 additions and 8 deletions

View File

@ -1,4 +1,5 @@
{ {
"_name_or_path": "opt-1.3b",
"activation_dropout": 0.0, "activation_dropout": 0.0,
"activation_function": "relu", "activation_function": "relu",
"architectures": [ "architectures": [
@ -6,23 +7,22 @@
], ],
"attention_dropout": 0.0, "attention_dropout": 0.0,
"bos_token_id": 2, "bos_token_id": 2,
"hidden_size": 2048,
"do_layer_norm_before": true, "do_layer_norm_before": true,
"dropout": 0.1, "dropout": 0.1,
"eos_token_id": 2, "eos_token_id": 2,
"ffn_dim": 8192, "ffn_dim": 8192,
"hidden_size": 2048,
"init_std": 0.02, "init_std": 0.02,
"layerdrop": 0.0, "layerdrop": 0.0,
"max_position_embeddings": 2048, "max_position_embeddings": 2048,
"model_type": "opt",
"num_attention_heads": 32, "num_attention_heads": 32,
"num_hidden_layers": 24, "num_hidden_layers": 24,
"pad_token_id": 1, "pad_token_id": 1,
"prefix": "</s>",
"torch_dtype": "float16", "torch_dtype": "float16",
"transformers_version": "4.19.0.dev0", "transformers_version": "4.20.0.dev0",
"use_cache": true, "use_cache": true,
"vocab_size": 50272, "vocab_size": 50272,
"word_embed_proj_dim": 2048, "word_embed_proj_dim": 2048
"prefix": "</s>"
} }

View File

@ -1,3 +1,3 @@
version https://git-lfs.github.com/spec/v1 version https://git-lfs.github.com/spec/v1
oid sha256:cab5d1b7b11900213091184b559d3201ddaeaa4a2c12bef4ae14a90ceee7113c oid sha256:774f854fb3f9f6723cfec007648e34f33457144ebdbd0b361a6edbef2777f371
size 5263424312 size 2631916344