From 5a87146c00d58ecbb678628fe7dea411682ad606 Mon Sep 17 00:00:00 2001
From: bojiang <bojiang_@outlook.com>
Date: Fri, 27 Sep 2024 20:18:00 +0800
Subject: [PATCH] doc: adjust model priority

---
 gen_readme.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gen_readme.py b/gen_readme.py
index 04ab616c..b7d18596 100644
--- 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"],
                 }
             )