Text Generation
Transformers
PyTorch
Safetensors
bloom
text-generation-inference
Inference Endpoints
Muennighoff lewtun HF staff commited on
Commit
8ba1b1c
1 Parent(s): 952637e

Fix architecture (#1)

Browse files

- Fix architecture (04363a9e07cdc4d0517386760934c8b6dd966c2a)


Co-authored-by: Lewis Tunstall <lewtun@users.noreply.huggingface.co>

Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "apply_residual_connection_post_layernorm": false,
3
  "architectures": [
4
- "BloomModel"
5
  ],
6
  "attention_dropout": 0.0,
7
  "attention_softmax_in_fp32": true,
@@ -28,4 +28,4 @@
28
  "unk_token_id": 0,
29
  "use_cache": true,
30
  "vocab_size": 250880
31
- }
 
1
  {
2
  "apply_residual_connection_post_layernorm": false,
3
  "architectures": [
4
+ "BloomForCausalLM"
5
  ],
6
  "attention_dropout": 0.0,
7
  "attention_softmax_in_fp32": true,
 
28
  "unk_token_id": 0,
29
  "use_cache": true,
30
  "vocab_size": 250880
31
+ }