Text Generation
Transformers
PyTorch
Safetensors
bloom
Eval Results
text-generation-inference
Inference Endpoints
cakiki lewtun HF staff commited on
Commit
3dfc1e0
1 Parent(s): c0f6835

Fix architecture (#35)

Browse files

- Fix architecture (0dbd8776f6ca4e35a8e75d21997b52c0dc2911fc)


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

Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -4,7 +4,7 @@
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,
@@ -26,4 +26,4 @@
26
  "transformers_version": "4.20.0",
27
  "use_cache": true,
28
  "vocab_size": 250880
29
- }
 
4
  "attention_softmax_in_fp32": true,
5
  "bias_dropout_fusion": true,
6
  "architectures": [
7
+ "BloomForCausalLM"
8
  ],
9
  "bos_token_id": 1,
10
  "eos_token_id": 2,
 
26
  "transformers_version": "4.20.0",
27
  "use_cache": true,
28
  "vocab_size": 250880
29
+ }