From 8040ae3a4ab875eb6595ce5be158b39e81a71b2b Mon Sep 17 00:00:00 2001
From: Iason Myttas <iasonm@graphcore.ai>
Date: Tue, 28 Jan 2025 15:01:39 +0000
Subject: [PATCH] Changing llama_v2_70b to correct config class

---
 torchbenchmark/util/framework/huggingface/basic_configs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchbenchmark/util/framework/huggingface/basic_configs.py b/torchbenchmark/util/framework/huggingface/basic_configs.py
index f0941f7df2..4b66a662d2 100644
--- a/torchbenchmark/util/framework/huggingface/basic_configs.py
+++ b/torchbenchmark/util/framework/huggingface/basic_configs.py
@@ -135,7 +135,7 @@
         512,
         512,
         'AutoConfig.from_pretrained("meta-llama/Llama-2-70b-hf")',
-        "AutoModelForMaskedLM",
+        "AutoModelForCausalLM",
     ),
     "llama_v31_8b": (
         512,