bigscience
/

distill-bloom-1b3-10x

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

lewtun HF staff commited on Feb 10, 2023

Commit

a00218b

•

1 Parent(s): 4a8f399

Fix architecture

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "apply_residual_connection_post_layernorm": false,
   "architectures": [
-    "BloomModel"
   ],
   "attention_dropout": 0.0,
   "attention_softmax_in_fp32": true,
@@ -32,4 +32,4 @@
   "vocab_size": 250880,
   "weights_aggregation_strategy": "mean",
   "width_downsampling_rate": 0.25
-}

 {
   "apply_residual_connection_post_layernorm": false,
   "architectures": [
+    "BloomForCausalLM"
   ],
   "attention_dropout": 0.0,
   "attention_softmax_in_fp32": true,
   "vocab_size": 250880,
   "weights_aggregation_strategy": "mean",
   "width_downsampling_rate": 0.25
+}