Younes Belkada commited on
Commit
c33bc30
1 Parent(s): 9b0b057

add architecture

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