cakiki lewtun HF staff commited on
Commit
226c467
1 Parent(s): 9c4abbe

Fix architecture (#39)

Browse files

- Fix architecture (7a3f23a1111551b168699d5f01d4ec9143419b4b)


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

Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -3,7 +3,7 @@
3
  "attention_dropout": 0.0,
4
  "attention_softmax_in_fp32": true,
5
  "architectures": [
6
- "BloomModel"
7
  ],
8
  "bias_dropout_fusion": true,
9
  "bos_token_id": 1,
@@ -27,4 +27,4 @@
27
  "transformers_version": "4.20.0",
28
  "use_cache": true,
29
  "vocab_size": 250880
30
- }
 
3
  "attention_dropout": 0.0,
4
  "attention_softmax_in_fp32": true,
5
  "architectures": [
6
+ "BloomForCausalLM"
7
  ],
8
  "bias_dropout_fusion": true,
9
  "bos_token_id": 1,
 
27
  "transformers_version": "4.20.0",
28
  "use_cache": true,
29
  "vocab_size": 250880
30
+ }