Update src/display/utils.py
Browse files- src/display/utils.py +0 -21
src/display/utils.py
CHANGED
@@ -61,27 +61,6 @@ class ModelDetails:
|
|
61 |
symbol: str = "" # emoji
|
62 |
|
63 |
|
64 |
-
# class ModelType(Enum):
|
65 |
-
# PT = ModelDetails(name="pretrained", symbol="🟢")
|
66 |
-
# FT = ModelDetails(name="fine-tuned", symbol="🔶")
|
67 |
-
# IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
|
68 |
-
# RL = ModelDetails(name="RL-tuned", symbol="🟦")
|
69 |
-
# Unknown = ModelDetails(name="", symbol="?")
|
70 |
-
|
71 |
-
# def to_str(self, separator=" "):
|
72 |
-
# return f"{self.value.symbol}{separator}{self.value.name}"
|
73 |
-
|
74 |
-
# @staticmethod
|
75 |
-
# def from_str(type):
|
76 |
-
# if "fine-tuned" in type or "🔶" in type:
|
77 |
-
# return ModelType.FT
|
78 |
-
# if "pretrained" in type or "🟢" in type:
|
79 |
-
# return ModelType.PT
|
80 |
-
# if "RL-tuned" in type or "🟦" in type:
|
81 |
-
# return ModelType.RL
|
82 |
-
# if "instruction-tuned" in type or "⭕" in type:
|
83 |
-
# return ModelType.IFT
|
84 |
-
# return ModelType.Unknown
|
85 |
class ModelType(Enum):
|
86 |
PT = ModelDetails(name="pretrained", symbol="🟢")
|
87 |
CPT = ModelDetails(name="continuously pretrained", symbol="🟩")
|
|
|
61 |
symbol: str = "" # emoji
|
62 |
|
63 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
64 |
class ModelType(Enum):
|
65 |
PT = ModelDetails(name="pretrained", symbol="🟢")
|
66 |
CPT = ModelDetails(name="continuously pretrained", symbol="🟩")
|