Fix architecture (#36)

- Fix architecture (969bcd1df9947295f1293b907b977e601d4f8fbd)


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

config.json

@@ -5,7 +5,7 @@
   "bias_dropout_fusion": true,
   "bos_token_id": 1,
   "architectures": [
-    "BloomModelForCausalLM"
+    "BloomForCausalLM"
   ],
   "eos_token_id": 2,
   "pad_token_id": 3,
@@ -26,4 +26,4 @@
   "transformers_version": "4.20.0",
   "use_cache": true,
   "vocab_size": 250880
-}
+}