bigscience
/

bloom-560m

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Slightly modify model card

#1

by ybelkada - opened May 30, 2022

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -122,11 +122,11 @@ Please see [the BLOOM training README](https://github.com/bigscience-workshop/bi
 * ALiBI positional encodings (see [paper](https://arxiv.org/pdf/2108.12409.pdf)), with GeLU activation functions
-* 176 billion parameters:
-    * 70 layers, 112 attention heads
-    * Hidden layers are 14336-dimensional
     * Sequence length of 2048 tokens used (see [BLOOM tokenizer](https://huggingface.co/bigscience/tokenizer), [tokenizer description](#tokenization))

 * ALiBI positional encodings (see [paper](https://arxiv.org/pdf/2108.12409.pdf)), with GeLU activation functions
+* 350 million parameters:
+    * 24 layers, 16 attention heads
+    * Hidden layers are 1024-dimensional
     * Sequence length of 2048 tokens used (see [BLOOM tokenizer](https://huggingface.co/bigscience/tokenizer), [tokenizer description](#tokenization))