tathagataraha commited on
Commit
e1cdc4b
Β·
1 Parent(s): 671e1a6

[FIX] Preference tuned model symbol

Browse files
Files changed (1) hide show
  1. src/display/utils.py +2 -2
src/display/utils.py CHANGED
@@ -95,8 +95,8 @@ class ModelType(Enum):
95
  # return ModelType.FT
96
  if "pretrained" in type or "🟒" in type:
97
  return ModelType.PT
98
- # if "RL-tuned" in type or "🟦" in type:
99
- # return ModelType.RL
100
  if "instruction-tuned" in type or "β­•" in type:
101
  return ModelType.IFT
102
  # if "domain-specific" in type or "βž•" in type:
 
95
  # return ModelType.FT
96
  if "pretrained" in type or "🟒" in type:
97
  return ModelType.PT
98
+ if "preference-tuned" in type or "🟦" in type:
99
+ return ModelType.RL
100
  if "instruction-tuned" in type or "β­•" in type:
101
  return ModelType.IFT
102
  # if "domain-specific" in type or "βž•" in type: