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

Update README.md and utils.py

Browse files
Files changed (1) hide show
  1. src/display/utils.py +5 -0
src/display/utils.py CHANGED
@@ -66,6 +66,7 @@ class ModelType(Enum):
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=" "):
@@ -81,6 +82,10 @@ class ModelType(Enum):
81
  return ModelType.RL
82
  if "instruction-tuned" in type or "⭕" in type:
83
  return ModelType.IFT
 
 
 
 
84
  return ModelType.Unknown
85
 
86
  class WeightType(Enum):
 
66
  FT = ModelDetails(name="fine-tuned", symbol="🔶")
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=" "):
 
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
+
89
  return ModelType.Unknown
90
 
91
  class WeightType(Enum):