From 3f9c5b32fccc6ab184f22aa875a3876ebbe9773e Mon Sep 17 00:00:00 2001
From: Rick Zhou <rickzhoucmu@gmail.com>
Date: Tue, 23 Jul 2024 18:13:42 -0700
Subject: [PATCH] Update gen_readme.py

---
 gen_readme.py | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/gen_readme.py b/gen_readme.py
index e9af55d9..906d3c14 100644
--- a/gen_readme.py
+++ b/gen_readme.py
@@ -8,6 +8,7 @@
 grouped_data = defaultdict(list)
 
 model_display_names = {
+    "llama3.1": "Llama-3.1",
     "llama3": "Llama-3",
     "phi3": "Phi-3",
     "mistral": "Mistral",
@@ -18,13 +19,14 @@
 }
 
 model_priority = {
-    "llama3": 1,
-    "phi3": 2,
-    "mistral": 3,
-    "qwen2": 4,
-    "gemma": 5,
-    "llama2": 6,
-    "mixtral": 7,
+    "llama3.1": 1,
+    "llama3": 2,
+    "phi3": 3,
+    "mistral": 4,
+    "qwen2": 5,
+    "gemma": 6,
+    "llama2": 7,
+    "mixtral": 8,
 }
 
 yaml_files = glob.glob(os.path.join(root_dir, "**/bento.yaml"), recursive=True)