flax_model.msgpack.index.json

{
  "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"
  }
}