モデルタイプの追加・変更: マルチモーダルとベースマージを追加し、RL-tunedの名称を更新

src/display/utils.py

@@ -104,7 +104,9 @@ class ModelType(Enum):
     PT = ModelDetails(name="pretrained", symbol="🟢")
     FT = ModelDetails(name="fine-tuned", symbol="🔶")
     IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
-    RL = ModelDetails(name="RL-tuned", symbol="🟦")
+    RL = ModelDetails(name="RL-tuned (Preference optimization)", symbol="🟦")
+    MM = ModelDetails(name="multimodal", symbol="🌟")
+    BM = ModelDetails(name="base merges", symbol="💫")
 
     def to_str(self, separator=" "):
         return f"{self.value.symbol}{separator}{self.value.name}"
@@ -119,6 +121,10 @@ class ModelType(Enum):
             return ModelType.RL
         if "instruction-tuned" in type or "⭕" in type:
             return ModelType.IFT
+        if "multimodal" in type or "🌟" in type:
+            return ModelType.MM
+        if "base merges" in type or "💫" in type:
+            return ModelType.BM
         raise ValueError(f"Unsupported model type: {type}")