Text Generation
Transformers
PyTorch
Safetensors
bloom
text-generation-inference
Inference Endpoints

Fix architecture

#1
by lewtun HF staff - opened
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "apply_residual_connection_post_layernorm": false,
3
  "architectures": [
4
- "BloomModel"
5
  ],
6
  "attention_dropout": 0.0,
7
  "attention_softmax_in_fp32": true,
@@ -32,4 +32,4 @@
32
  "vocab_size": 250880,
33
  "weights_aggregation_strategy": "mean",
34
  "width_downsampling_rate": 0.25
35
- }
 
1
  {
2
  "apply_residual_connection_post_layernorm": false,
3
  "architectures": [
4
+ "BloomForCausalLM"
5
  ],
6
  "attention_dropout": 0.0,
7
  "attention_softmax_in_fp32": true,
 
32
  "vocab_size": 250880,
33
  "weights_aggregation_strategy": "mean",
34
  "width_downsampling_rate": 0.25
35
+ }