Fix architecture

#5
by lewtun HF staff - opened
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -5,7 +5,7 @@
5
  "bias_dropout_fusion": true,
6
  "bos_token_id": 1,
7
  "architectures": [
8
- "BloomModel"
9
  ],
10
  "eos_token_id": 2,
11
  "pad_token_id": 3,
 
5
  "bias_dropout_fusion": true,
6
  "bos_token_id": 1,
7
  "architectures": [
8
+ "BloomModelForCausalLM"
9
  ],
10
  "eos_token_id": 2,
11
  "pad_token_id": 3,