diff --git a/examples/llama-3/lora-8b-tp.yml b/examples/llama-3/lora-8b-tp.yml
index fdf90bb33..628f42736 100644
--- a/examples/llama-3/lora-8b-tp.yml
+++ b/examples/llama-3/lora-8b-tp.yml
@@ -1,4 +1,4 @@
-base_model: NousResearch/Meta-Llama-3-8B
+base_model: NousResearch/Meta-Llama-3.1-8B
 model_type: LlamaForCausalLM
 tokenizer_type: AutoTokenizer