Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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="π©")
|