lewtun HF staff commited on
Commit
9627ca1
1 Parent(s): ceb61fa

Fix architecture

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -4,7 +4,7 @@
4
  "attention_softmax_in_fp32": true,
5
  "bias_dropout_fusion": true,
6
  "architectures": [
7
- "BloomModel"
8
  ],
9
  "bos_token_id": 1,
10
  "eos_token_id": 2,
 
4
  "attention_softmax_in_fp32": true,
5
  "bias_dropout_fusion": true,
6
  "architectures": [
7
+ "BloomModelForCausalLM"
8
  ],
9
  "bos_token_id": 1,
10
  "eos_token_id": 2,