choco9966 commited on
Commit
4f7484b
β€’
1 Parent(s): e40c23f

fix symbole

Browse files
Files changed (1) hide show
  1. src/display/utils.py +6 -6
src/display/utils.py CHANGED
@@ -84,12 +84,12 @@ class ModelDetails:
84
 
85
 
86
  class ModelType(Enum):
87
- PT = ModelDetails(name="🟒 pretrained", symbol="🟒")
88
- CPT = ModelDetails(name="🟩 continuously pretrained", symbol="🟩")
89
- FT = ModelDetails(name="πŸ”Ά fine-tuned on domain-specific datasets", symbol="πŸ”Ά")
90
- chat = ModelDetails(name="πŸ’¬ chat models (RLHF, DPO, IFT, ...)", symbol="πŸ’¬")
91
- merges = ModelDetails(name="🀝 base merges and moerges", symbol="🀝")
92
- Unknown = ModelDetails(name="❓ other", symbol="❓")
93
 
94
  def to_str(self, separator=" "):
95
  return f"{self.value.symbol}{separator}{self.value.name}"
 
84
 
85
 
86
  class ModelType(Enum):
87
+ PT = ModelDetails(name="pretrained", symbol="🟒")
88
+ CPT = ModelDetails(name="continuously pretrained", symbol="🟩")
89
+ FT = ModelDetails(name="fine-tuned on domain-specific datasets", symbol="πŸ”Ά")
90
+ chat = ModelDetails(name="chat models (RLHF, DPO, IFT, ...)", symbol="πŸ’¬")
91
+ merges = ModelDetails(name="base merges and moerges", symbol="🀝")
92
+ Unknown = ModelDetails(name="other", symbol="❓")
93
 
94
  def to_str(self, separator=" "):
95
  return f"{self.value.symbol}{separator}{self.value.name}"