Upload BloomForCausalLM

config.json

@@ -4,21 +4,29 @@
     "BloomForCausalLM"
   ],
   "attention_dropout": 0.0,
+  "attention_softmax_in_fp32": true,
+  "bias_dropout_fusion": true,
   "bos_token_id": 1,
+  "dtype": "float32",
   "eos_token_id": 2,
   "hidden_dropout": 0.0,
   "hidden_size": 8,
   "initializer_range": 0.02,
-  "intermediate_size": 32,
   "layer_norm_epsilon": 1e-05,
+  "masked_softmax_fusion": true,
   "model_type": "bloom",
   "n_head": 4,
+  "n_inner": null,
   "n_layer": 2,
-  "num_key_value_heads": 2,
+  "offset_alibi": 100,
+  "pad_token_id": 3,
   "pretraining_tp": 1,
+  "seq_length": 2048,
+  "skip_bias_add": true,
+  "skip_bias_add_qkv": false,
   "slow_but_exact": false,
-  "torch_dtype": "float32",
-  "transformers_version": "4.55.0.dev0",
+  "transformers_version": "4.57.0.dev0",
+  "unk_token_id": 0,
   "use_cache": true,
-  "vocab_size": 250684
+  "vocab_size": 250880
 }

generation_config.json

@@ -2,5 +2,6 @@
   "_from_model_config": true,
   "bos_token_id": 1,
   "eos_token_id": 2,
-  "transformers_version": "4.55.0.dev0"
+  "pad_token_id": 3,
+  "transformers_version": "4.57.0.dev0"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0eeaaa977f353edc41a4e58802c22a2e0a9445d2d394f63eae06ae0aa1a11ee3
-size 8031936
+oid sha256:b8d07ab4eddd24e632d1bf750148545127f4f9da4f676d759c95cb7526e9bcbb
+size 8038208