From 520a4b10ae8ee426934ee222cbc905d1aa9c6319 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Wed, 5 Mar 2025 19:57:13 +0800
Subject: [PATCH] Update determine_base_model function to return version_string
 and add TODO for more mappings

---
 py/utils/model_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/py/utils/model_utils.py b/py/utils/model_utils.py
index 9c09e00b..80c2d023 100644
--- a/py/utils/model_utils.py
+++ b/py/utils/model_utils.py
@@ -23,4 +23,5 @@ def determine_base_model(version_string: Optional[str]) -> str:
         if key in version_lower:
             return value
     
-    return "Unknown" 
\ No newline at end of file
+    # TODO: Add more base model mappings
+    return version_string 
\ No newline at end of file