Fix architecture (#1)

- Fix architecture (0837dcca56aff5aab40a8b9abe94d6e0c6af137b)


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

config.json

@@ -2,7 +2,7 @@
   "_name_or_path": "/gpfsscratch/rech/six/commun/commun/experiments/muennighoff/bloomckpt/6b3/bloom-7b1",
   "apply_residual_connection_post_layernorm": false,
   "architectures": [
-    "BloomModel"
+    "BloomForCausalLM"
   ],
   "attention_dropout": 0.0,
   "attention_softmax_in_fp32": true,
@@ -30,4 +30,4 @@
   "unk_token_id": 0,
   "use_cache": true,
   "vocab_size": 250682
-}
+}