bigscience
/

bloomz-3b

lewtun HF staff commited on Feb 10, 2023

Commit

d08d11f

•

1 Parent(s): ceb61fa

Fix architecture (#6)

- Fix architecture (9627ca11a6723d600048f54067f2e21b8b0dc606)

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

Files changed (1) hide show

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,
   "architectures": [
-    "BloomModel"
   ],
   "bos_token_id": 1,
   "eos_token_id": 2,

   "attention_softmax_in_fp32": true,
   "bias_dropout_fusion": true,
   "architectures": [
+    "BloomModelForCausalLM"
   ],
   "bos_token_id": 1,
   "eos_token_id": 2,