bigscience
/

bloom-560m

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

lewtun HF staff commited on Feb 10, 2023

Commit

cd69902

•

1 Parent(s): afe2e6f

Fix architecture

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "apply_residual_connection_post_layernorm": false,
   "attention_dropout": 0.0,
   "architectures": [
-    "BloomModel"
   ],
   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,

   "apply_residual_connection_post_layernorm": false,
   "attention_dropout": 0.0,
   "architectures": [
+    "BloomModelForCausalLM"
   ],
   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,