File size: 1,742 Bytes
3c938d6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
{
"metadata": {
"total_size": 14138032128
},
"weight_map": {
"transformer/h/FlaxBloomBlockLayers/input_layernorm/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/input_layernorm/scale": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/mlp/dense_4h_to_h/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/mlp/dense_4h_to_h/kernel": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/mlp/dense_h_to_4h/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/mlp/dense_h_to_4h/kernel": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/post_attention_layernorm/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/post_attention_layernorm/scale": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/self_attention/dense/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/self_attention/dense/kernel": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/self_attention/query_key_value/bias": "flax_model-00001-of-00002.msgpack",
"transformer/h/FlaxBloomBlockLayers/self_attention/query_key_value/kernel": "flax_model-00002-of-00002.msgpack",
"transformer/ln_f/bias": "flax_model-00002-of-00002.msgpack",
"transformer/ln_f/scale": "flax_model-00002-of-00002.msgpack",
"transformer/word_embeddings/embedding": "flax_model-00002-of-00002.msgpack",
"transformer/word_embeddings_layernorm/bias": "flax_model-00002-of-00002.msgpack",
"transformer/word_embeddings_layernorm/scale": "flax_model-00002-of-00002.msgpack"
}
}
|