open_pl_llm_leaderboard

Restarting on CPU Upgrade

djstrong commited on Mar 19

Commit

68bbe4a

•

1 Parent(s): b5aa7e1

baseline type

Files changed (1) hide show

src/display/utils.py CHANGED Viewed

@@ -90,7 +90,7 @@ class ModelType(Enum):
         if "instruction-tuned" in type or "⭕" in type:
             return ModelType.IFT
         if "baseline" in type or "⚖" in type:
-            return ModelType.IFT
         return ModelType.Unknown
 class WeightType(Enum):

         if "instruction-tuned" in type or "⭕" in type:
             return ModelType.IFT
         if "baseline" in type or "⚖" in type:
+            return ModelType.Baseline
         return ModelType.Unknown
 class WeightType(Enum):