bigscience
/

bloom-1b7

Text Generation

text-generation-inference

Model card Files Files and versions

lewtun HF Staff commited on Feb 10, 2023

Commit

7a3f23a

·

1 Parent(s): 9c4abbe

Fix architecture

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -3,7 +3,7 @@
   "attention_dropout": 0.0,
   "attention_softmax_in_fp32": true,
   "architectures": [
-    "BloomModel"
   ],
   "bias_dropout_fusion": true,
   "bos_token_id": 1,
@@ -27,4 +27,4 @@
   "transformers_version": "4.20.0",
   "use_cache": true,
   "vocab_size": 250880
-}

   "attention_dropout": 0.0,
   "attention_softmax_in_fp32": true,
   "architectures": [
+    "BloomForCausalLM"
   ],
   "bias_dropout_fusion": true,
   "bos_token_id": 1,
   "transformers_version": "4.20.0",
   "use_cache": true,
   "vocab_size": 250880
+}