multimodal, base merges

src/display/utils.py

@@ -105,8 +105,8 @@ class ModelType(Enum):
     FT = ModelDetails(name="fine-tuned", symbol="🔶")
     IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
     RL = ModelDetails(name="RL-tuned (Preference optimization)", symbol="🟦")
-    MM = ModelDetails(name="multimodal", symbol="🌟")
-    BM = ModelDetails(name="base merges", symbol="💫")
+    MM = ModelDetails(name="multimodal", symbol="🌸")
+    BM = ModelDetails(name="base merges and moerges", symbol="🤝")
 
     def to_str(self, separator=" "):
         return f"{self.value.symbol}{separator}{self.value.name}"
@@ -121,9 +121,9 @@ class ModelType(Enum):
             return ModelType.RL
         if "instruction-tuned" in type or "⭕" in type:
             return ModelType.IFT
-        if "multimodal" in type or "🌟" in type:
+        if "multimodal" in type or "🌸" in type:
             return ModelType.MM
-        if "base merges" in type or "💫" in type:
+        if "base merges and moerges" in type or "🤝" in type:
             return ModelType.BM
         raise ValueError(f"Unsupported model type: {type}")