update logo

src/display/utils.py

@@ -148,9 +148,9 @@ class ModelDetails:
 class ModelType(Enum):
     PT = ModelDetails(name="GPTQ", symbol="🟢")
     CPT = ModelDetails(name="AWQ", symbol="🟩")
-    FT = ModelDetails(name="llama.cpp", symbol="🔶")
-    chat = ModelDetails(name="Bisandbytes", symbol="💬")
-    merges = ModelDetails(name="AutoRound", symbol="🌐")
+    FT = ModelDetails(name="llama.cpp", symbol="🔷")
+    chat = ModelDetails(name="Bisandbytes", symbol="🔵")
+    merges = ModelDetails(name="AutoRound", symbol="🍒")
     Unknown = ModelDetails(name="", symbol="?")
 
     def to_str(self, separator=" "):
@@ -158,15 +158,15 @@ class ModelType(Enum):
 
     @staticmethod
     def from_str(type):
-        if "fine-tuned" in type or "🔶" in 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 any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "💬"]]):
+        if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "🔵"]]):
             return ModelType.chat
-        if "merge" in type or "🌐" in type:
+        if "merge" in type or "🍒" in type:
             return ModelType.merges
         return ModelType.Unknown
 """
@@ -174,9 +174,9 @@ class ModelType(Enum):
 class ModelType(Enum):
     PT = ModelDetails(name="pretrained", symbol="🟢")
     CPT = ModelDetails(name="continuously pretrained", symbol="🟩")
-    FT = ModelDetails(name="fine-tuned on domain-specific datasets", symbol="🔶")
-    chat = ModelDetails(name="chat models (RLHF, DPO, IFT, ...)", symbol="💬")
-    merges = ModelDetails(name="base merges and moerges", symbol="🌐")
+    FT = ModelDetails(name="fine-tuned on domain-specific datasets", symbol="🔷")
+    chat = ModelDetails(name="chat models (RLHF, DPO, IFT, ...)", symbol="🔵")
+    merges = ModelDetails(name="base merges and moerges", symbol="🍒")
     Unknown = ModelDetails(name="", symbol="?")
 
     def to_str(self, separator=" "):
@@ -184,15 +184,15 @@ class ModelType(Enum):
 
     @staticmethod
     def from_str(type):
-        if "fine-tuned" in type or "🔶" in 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 or "quantization" in type:
             return ModelType.PT
-        if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "💬"]]):
+        if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "🔵"]]):
             return ModelType.chat
-        if "merge" in type or "🌐" in type:
+        if "merge" in type or "🍒" in type:
             return ModelType.merges
         return ModelType.Unknown
 
@@ -206,8 +206,8 @@ class QuantType(Enum):
     gptq = ModelDetails(name="GPTQ", symbol="🟢")
     aqlm = ModelDetails(name="AQLM", symbol="⭐")
     awq = ModelDetails(name="AWQ", symbol="🟩")
-    llama_cpp = ModelDetails(name="llama.cpp", symbol="🔶")
-    bnb = ModelDetails(name="bitsandbytes", symbol="💬")
+    llama_cpp = ModelDetails(name="llama.cpp", symbol="🔷")
+    bnb = ModelDetails(name="bitsandbytes", symbol="🔵")
     autoround = ModelDetails(name="AutoRound", symbol="🍒")
     Unknown = ModelDetails(name="?", symbol="?")