bigscience
/

bloom-560m

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

Fix architecture

#50

by lewtun HF staff - opened Feb 10, 2023

base: refs/heads/main

←

from: refs/pr/50

Discussion Files changed

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "apply_residual_connection_post_layernorm": false,
   "attention_dropout": 0.0,
   "architectures": [
-    "BloomModelForCausalLM"
   ],
   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,
@@ -26,4 +26,4 @@
   "transformers_version": "4.20.0",
   "use_cache": true,
   "vocab_size": 250880
-}

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