Fix architecture (#35)

- Fix architecture (0dbd8776f6ca4e35a8e75d21997b52c0dc2911fc)


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

config.json

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