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