doc: adjust model priority

bentoml · Sep 27, 2024 · 5a87146 · 5a87146
1 parent c32b111
commit 5a87146
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gen_readme.py b/gen_readme.py
@@ -26,11 +26,11 @@
 
 model_priority = {
     "llama3.2": 1,
-    "llama3.1": 2,
+    "qwen2.5": 2,
+    "pixtral": 2,
     "phi3": 3,
     "mistral": 4,
     "gemma2": 5,
-    "qwen2.5": 6,
     "mixtral": 7,
     "mistral-large": 8,
     "codestral": 9,
@@ -50,7 +50,7 @@
             grouped_data[data["name"]].append(
                 {
                     "name": data["name"],
-                    "version": data["version"],
+                    "version": data["version"].rsplit("-", 1)[0],
                     "model_name": data["labels"]["model_name"],
                 }
             )