Ali Elfilali commited on
Commit
fbb29de
β€’
1 Parent(s): 8b320de

Update src/display/utils.py

Browse files
Files changed (1) hide show
  1. 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="🟩")