Spaces:
Sleeping
Sleeping
Bazsalanszky
commited on
Commit
β’
d044810
1
Parent(s):
815d1ad
Switch emojis
Browse files- src/display/utils.py +4 -4
src/display/utils.py
CHANGED
@@ -65,9 +65,9 @@ class ModelDetails:
|
|
65 |
|
66 |
|
67 |
class ModelType(Enum):
|
68 |
-
PT = ModelDetails(name="pretrained", symbol="
|
69 |
FT = ModelDetails(name="fine-tuned", symbol="πΆ")
|
70 |
-
IFT = ModelDetails(name="instruction-tuned", symbol="
|
71 |
RL = ModelDetails(name="RL-tuned", symbol="π¦")
|
72 |
Unknown = ModelDetails(name="", symbol="?")
|
73 |
|
@@ -78,11 +78,11 @@ class ModelType(Enum):
|
|
78 |
def from_str(type):
|
79 |
if "fine-tuned" in type or "πΆ" in type:
|
80 |
return ModelType.FT
|
81 |
-
if "pretrained" in type or "
|
82 |
return ModelType.PT
|
83 |
if "RL-tuned" in type or "π¦" in type:
|
84 |
return ModelType.RL
|
85 |
-
if "instruction-tuned" in type or "
|
86 |
return ModelType.IFT
|
87 |
return ModelType.Unknown
|
88 |
|
|
|
65 |
|
66 |
|
67 |
class ModelType(Enum):
|
68 |
+
PT = ModelDetails(name="pretrained", symbol="β")
|
69 |
FT = ModelDetails(name="fine-tuned", symbol="πΆ")
|
70 |
+
IFT = ModelDetails(name="instruction-tuned", symbol="π’")
|
71 |
RL = ModelDetails(name="RL-tuned", symbol="π¦")
|
72 |
Unknown = ModelDetails(name="", symbol="?")
|
73 |
|
|
|
78 |
def from_str(type):
|
79 |
if "fine-tuned" in type or "πΆ" in type:
|
80 |
return ModelType.FT
|
81 |
+
if "pretrained" in type or "β" in type:
|
82 |
return ModelType.PT
|
83 |
if "RL-tuned" in type or "π¦" in type:
|
84 |
return ModelType.RL
|
85 |
+
if "instruction-tuned" in type or "π’" in type:
|
86 |
return ModelType.IFT
|
87 |
return ModelType.Unknown
|
88 |
|