sgugger lewtun HF staff commited on
Commit
cbc603c
1 Parent(s): b7c6845

Fix architecture class (#33)

Browse files

- Fix architecture class (4bc6f8387d1ace05308b76e7e12c669c078733de)


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

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -2,7 +2,7 @@
2
  "apply_residual_connection_post_layernorm": false,
3
  "attention_dropout": 0.0,
4
  "architectures": [
5
- "BloomModel"
6
  ],
7
  "attention_softmax_in_fp32": true,
8
  "seq_length": 2048,
 
2
  "apply_residual_connection_post_layernorm": false,
3
  "attention_dropout": 0.0,
4
  "architectures": [
5
+ "BloomModelForCausalLM"
6
  ],
7
  "attention_softmax_in_fp32": true,
8
  "seq_length": 2048,