cakiki lewtun HF staff commited on
Commit
4f161c3
1 Parent(s): cfba2fb

Fix architecture (#2)

Browse files

- Fix architecture (849e1c25fb817739f88aedb66cab48db40ace538)


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

Files changed (1) hide show
  1. config.json +2 -3
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
  "pad_token_id": 3,
@@ -21,5 +21,4 @@
21
  "transformers_version": "4.21.0",
22
  "use_cache": true,
23
  "vocab_size": 250880
24
- }
25
-
 
2
  "apply_residual_connection_post_layernorm": false,
3
  "attention_dropout": 0.0,
4
  "architectures": [
5
+ "BloomForCausalLM"
6
  ],
7
  "attention_softmax_in_fp32": true,
8
  "pad_token_id": 3,
 
21
  "transformers_version": "4.21.0",
22
  "use_cache": true,
23
  "vocab_size": 250880
24
+ }