zhuqiyuan commited on
Commit
5e21ab4
·
1 Parent(s): cdd9698

Update README.md and utils.py

Browse files
Files changed (1) hide show
  1. src/display/utils.py +2 -2
src/display/utils.py CHANGED
@@ -67,7 +67,7 @@ class ModelType(Enum):
67
  IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
68
  RL = ModelDetails(name="RL-tuned", symbol="🟦")
69
  LOCK = ModelDetails(name="locked", symbol="🔒")
70
- Unknown = ModelDetails(name="", symbol="?")
71
 
72
  def to_str(self, separator=" "):
73
  return f"{self.value.symbol}{separator}{self.value.name}"
@@ -82,7 +82,6 @@ class ModelType(Enum):
82
  return ModelType.RL
83
  if "instruction-tuned" in type or "⭕" in type:
84
  return ModelType.IFT
85
-
86
  if "locked" in type or "🔒" in type:
87
  return ModelType.LOCK
88
 
@@ -103,6 +102,7 @@ class Precision(Enum):
103
  return Precision.float16
104
  if precision in ["torch.bfloat16", "bfloat16"]:
105
  return Precision.bfloat16
 
106
  return Precision.Unknown
107
 
108
  # Column selection
 
67
  IFT = ModelDetails(name="instruction-tuned", symbol="⭕")
68
  RL = ModelDetails(name="RL-tuned", symbol="🟦")
69
  LOCK = ModelDetails(name="locked", symbol="🔒")
70
+ Unknown = ModelDetails(name="", symbol="🔒")
71
 
72
  def to_str(self, separator=" "):
73
  return f"{self.value.symbol}{separator}{self.value.name}"
 
82
  return ModelType.RL
83
  if "instruction-tuned" in type or "⭕" in type:
84
  return ModelType.IFT
 
85
  if "locked" in type or "🔒" in type:
86
  return ModelType.LOCK
87
 
 
102
  return Precision.float16
103
  if precision in ["torch.bfloat16", "bfloat16"]:
104
  return Precision.bfloat16
105
+
106
  return Precision.Unknown
107
 
108
  # Column selection