Hugging Face's logo

Spaces:
eduagarcia
/
open_pt_llm_leaderboard
Restarting on CPU Upgrade

App Files Community
12
eduagarcia commited on
Commit
155aef4
•
1 Parent(s): 395f537

Fix typo on ModelType

Browse files
Files changed (1) hide show
  1. src/display/utils.py +1 -1
src/display/utils.py CHANGED
@@ -179,7 +179,7 @@ class ModelType(Enum):
179
  if "fine-tuned" in type or "🔶" in type:
180
  return ModelType.FT
181
  if "language" in type or "🆎" in type:
182
- return ModelType.PT
183
  if "pretrained" in type or "🟢" in type:
184
  return ModelType.PT
185
  if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "💬"]]):
 
179
  if "fine-tuned" in type or "🔶" in type:
180
  return ModelType.FT
181
  if "language" in type or "🆎" in type:
182
+ return ModelType.LA
183
  if "pretrained" in type or "🟢" in type:
184
  return ModelType.PT
185
  if any([k in type for k in ["instruction-tuned", "RL-tuned", "chat", "🟦", "⭕", "💬"]]):