Update src/display/utils.py

src/display/utils.py

@@ -123,7 +123,8 @@ class ModelDetails:
 
 class ModelType(Enum):
     PT = ModelDetails(name="Pre trained", symbol="🟢")
-    SFT = ModelDetails(name="Supervised Finetuning", symbol="🔶")
+    FT = ModelDetails(name="Fine-tuned", symbol="🔶")
+    IFT = ModelDetails(name="Instruction-tuned", symbol="🔶")
     RL = ModelDetails(name="Reinforcement Learning", symbol="🟦")
     Unknown = ModelDetails(name="", symbol="?")
 
@@ -132,11 +133,13 @@ class ModelType(Enum):
 
     @staticmethod
     def from_str(type_str):
-        if "fine-tuned" in type_str or "instruction-tuned" in type_str or "Supervised Finetuning" in type_str or "🔶" in type_str or type_str == "SFT":
-            return ModelType.SFT
-        if "pretrained" in type_str or "Pre trained" in type_str or "🟢" in type_str or type_str == "PT":
+        if "fine-tuned" in type_str.lower() and "instruction" not in type_str.lower():
+            return ModelType.FT
+        if "instruction-tuned" in type_str.lower():
+            return ModelType.IFT
+        if "pretrained" in type_str.lower() or "pre trained" in type_str.lower() or "🟢" in type_str or type_str == "PT":
             return ModelType.PT
-        if "RL-tuned" in type_str or "Reinforcement Learning" in type_str or "🟦" in type_str or type_str == "RL":
+        if "rl-tuned" in type_str.lower() or "reinforcement learning" in type_str.lower() or "🟦" in type_str or type_str == "RL":
             return ModelType.RL
         return ModelType.Unknown