Update src/display/utils.py

src/display/utils.py

@@ -74,15 +74,15 @@ class ModelType(Enum):
 
     @staticmethod
     def from_str(type):
-        if "fine-tuned" in type or "🔶" in type:
-            return ModelType.FT
-        if "continously pretrained" in type or "🟩" in type:
-            return ModelType.CPT
         if "pretrained" in type or "🟢" in type:
             return ModelType.PT
+        if "continously pretrained" in type or "🟩" in type:
+            return ModelType.CPT
+        if "fine-tuned" in type or "🔶" in type:
+            return ModelType.FT
         if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "💬"]]):
             return ModelType.chat
-        if "merge" in type or "🤝" in type:
+        if "merges" in type or "🤝" in type:
             return ModelType.merges
         return ModelType.Unknown