Changing llama_v2_70b to correct config class

JasonMts · JasonMts · commit 8040ae3a4ab8 · 2025-02-07T14:53:24.000Z
diff --git 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,
