Update src/display/utils.py

src/display/utils.py

@@ -123,8 +123,7 @@ class ModelDetails:
 
 class ModelType(Enum):
     PT = ModelDetails(name="Pre trained", symbol="🟢")
-    FT = ModelDetails(name="Fine-tuned", symbol="🔶")
-    IFT = ModelDetails(name="Instruction-tuned", symbol="🔶")
+    SFT = ModelDetails(name="Supervised Finetuning", symbol="🔶")
     RL = ModelDetails(name="Reinforcement Learning", symbol="🟦")
     Unknown = ModelDetails(name="", symbol="?")
 
@@ -133,11 +132,13 @@ class ModelType(Enum):
 
     @staticmethod
     def from_str(type_str):
-        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":
+        if "fine-tuned" in type_str.lower() or \
+           "instruction-tuned" in type_str.lower() or \
+           "supervised finetuning" in type_str.lower() or \
+           "🔶" in type_str or \
+           type_str == "SFT" or type_str == "FT" or type_str == "IFT":
+            return ModelType.SFT
+        if "pretrained" in type_str.lower() or "pre trained" in type_str.lower() or "pré-treinado" in type_str.lower() or "🟢" in type_str or type_str == "PT":
             return ModelType.PT
         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