sanchit-gandhi commited on
Commit
3c938d6
·
1 Parent(s): 9d6ce34
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "apply_residual_connection_post_layernorm": false,
3
+ "architectures": [
4
+ "BloomForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "attention_softmax_in_fp32": true,
8
+ "bias_dropout_fusion": true,
9
+ "bos_token_id": 1,
10
+ "eos_token_id": 2,
11
+ "hidden_dropout": 0.0,
12
+ "initializer_range": 0.02,
13
+ "layer_norm_epsilon": 1e-05,
14
+ "masked_softmax_fusion": true,
15
+ "model_type": "bloom",
16
+ "n_embed": 4096,
17
+ "n_inner": null,
18
+ "n_layer": 30,
19
+ "num_attention_heads": 32,
20
+ "offset_alibi": 100,
21
+ "pad_token_id": 3,
22
+ "pretraining_tp": 4,
23
+ "seq_length": 2048,
24
+ "skip_bias_add": true,
25
+ "skip_bias_add_qkv": false,
26
+ "slow_but_exact": false,
27
+ "transformers_version": "4.21.0.dev0",
28
+ "unk_token_id": 0,
29
+ "use_cache": true,
30
+ "vocab_size": 250880
31
+ }
flax_model-00001-of-00002.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:614feec03366194304621bc18264a5f7ef4c4d5d1adfc40368fc21f9ae7870f1
3
+ size 9062892305
flax_model-00002-of-00002.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71219379745f4edec7284d87d06359f4e4a84cd0b0aa5b06dc7e4184ec371317
3
+ size 5075141103
flax_model.msgpack.index.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 14138032128
4
+ },
5
+ "weight_map": {
6
+ "transformer/h/FlaxBloomBlockLayers/input_layernorm/bias": "flax_model-00001-of-00002.msgpack",
7
+ "transformer/h/FlaxBloomBlockLayers/input_layernorm/scale": "flax_model-00001-of-00002.msgpack",
8
+ "transformer/h/FlaxBloomBlockLayers/mlp/dense_4h_to_h/bias": "flax_model-00001-of-00002.msgpack",
9
+ "transformer/h/FlaxBloomBlockLayers/mlp/dense_4h_to_h/kernel": "flax_model-00001-of-00002.msgpack",
10
+ "transformer/h/FlaxBloomBlockLayers/mlp/dense_h_to_4h/bias": "flax_model-00001-of-00002.msgpack",
11
+ "transformer/h/FlaxBloomBlockLayers/mlp/dense_h_to_4h/kernel": "flax_model-00001-of-00002.msgpack",
12
+ "transformer/h/FlaxBloomBlockLayers/post_attention_layernorm/bias": "flax_model-00001-of-00002.msgpack",
13
+ "transformer/h/FlaxBloomBlockLayers/post_attention_layernorm/scale": "flax_model-00001-of-00002.msgpack",
14
+ "transformer/h/FlaxBloomBlockLayers/self_attention/dense/bias": "flax_model-00001-of-00002.msgpack",
15
+ "transformer/h/FlaxBloomBlockLayers/self_attention/dense/kernel": "flax_model-00001-of-00002.msgpack",
16
+ "transformer/h/FlaxBloomBlockLayers/self_attention/query_key_value/bias": "flax_model-00001-of-00002.msgpack",
17
+ "transformer/h/FlaxBloomBlockLayers/self_attention/query_key_value/kernel": "flax_model-00002-of-00002.msgpack",
18
+ "transformer/ln_f/bias": "flax_model-00002-of-00002.msgpack",
19
+ "transformer/ln_f/scale": "flax_model-00002-of-00002.msgpack",
20
+ "transformer/word_embeddings/embedding": "flax_model-00002-of-00002.msgpack",
21
+ "transformer/word_embeddings_layernorm/bias": "flax_model-00002-of-00002.msgpack",
22
+ "transformer/word_embeddings_layernorm/scale": "flax_model-00002-of-00002.msgpack"
23
+ }
24
+ }