Fix architecture (#2)

- Fix architecture (849e1c25fb817739f88aedb66cab48db40ace538)


Co-authored-by: Lewis Tunstall <[email protected]>

config.json

@@ -2,7 +2,7 @@
   "apply_residual_connection_post_layernorm": false,
   "attention_dropout": 0.0,
   "architectures": [
-    "BloomModel"
+    "BloomForCausalLM"
   ],
   "attention_softmax_in_fp32": true,
   "pad_token_id": 3,
@@ -21,5 +21,4 @@
   "transformers_version": "4.21.0",
   "use_cache": true,
   "vocab_size": 250880
-}
-
+}