haoyang
commited on
Commit
β’
16b1a0b
1
Parent(s):
4b33d2c
update model metadata
Browse files- scripts/create_request_file.py +1 -1
- src/display/utils.py +6 -12
- src/leaderboard/read_evals.py +0 -2
scripts/create_request_file.py
CHANGED
@@ -12,7 +12,7 @@ EVAL_REQUESTS_PATH = "eval-queue"
|
|
12 |
QUEUE_REPO = "open-llm-leaderboard/requests"
|
13 |
|
14 |
precisions = ("float16", "bfloat16", "8bit (LLM.int8)", "4bit (QLoRA / FP4)", "GPTQ")
|
15 |
-
model_types = ("
|
16 |
weight_types = ("Original", "Delta", "Adapter")
|
17 |
|
18 |
|
|
|
12 |
QUEUE_REPO = "open-llm-leaderboard/requests"
|
13 |
|
14 |
precisions = ("float16", "bfloat16", "8bit (LLM.int8)", "4bit (QLoRA / FP4)", "GPTQ")
|
15 |
+
model_types = ("open-source", "close-source")
|
16 |
weight_types = ("Original", "Delta", "Adapter")
|
17 |
|
18 |
|
src/display/utils.py
CHANGED
@@ -65,10 +65,8 @@ class ModelDetails:
|
|
65 |
|
66 |
|
67 |
class ModelType(Enum):
|
68 |
-
|
69 |
-
|
70 |
-
IFT = ModelDetails(name="instruction-tuned", symbol="β")
|
71 |
-
RL = ModelDetails(name="RL-tuned", symbol="π¦")
|
72 |
Unknown = ModelDetails(name="", symbol="?")
|
73 |
|
74 |
def to_str(self, separator=" "):
|
@@ -76,14 +74,10 @@ class ModelType(Enum):
|
|
76 |
|
77 |
@staticmethod
|
78 |
def from_str(type):
|
79 |
-
if "
|
80 |
-
return ModelType.
|
81 |
-
if "
|
82 |
-
return ModelType.
|
83 |
-
if "RL-tuned" in type or "π¦" in type:
|
84 |
-
return ModelType.RL
|
85 |
-
if "instruction-tuned" in type or "β" in type:
|
86 |
-
return ModelType.IFT
|
87 |
return ModelType.Unknown
|
88 |
|
89 |
class WeightType(Enum):
|
|
|
65 |
|
66 |
|
67 |
class ModelType(Enum):
|
68 |
+
OSM = ModelDetails(name="open-source", symbol="π’")
|
69 |
+
CSM = ModelDetails(name="close-source", symbol="πΆ")
|
|
|
|
|
70 |
Unknown = ModelDetails(name="", symbol="?")
|
71 |
|
72 |
def to_str(self, separator=" "):
|
|
|
74 |
|
75 |
@staticmethod
|
76 |
def from_str(type):
|
77 |
+
if "close-source" in type or "πΆ" in type:
|
78 |
+
return ModelType.CSM
|
79 |
+
if "open-source" in type or "π’" in type:
|
80 |
+
return ModelType.OSM
|
|
|
|
|
|
|
|
|
81 |
return ModelType.Unknown
|
82 |
|
83 |
class WeightType(Enum):
|
src/leaderboard/read_evals.py
CHANGED
@@ -64,8 +64,6 @@ class EvalResult:
|
|
64 |
architectures = getattr(model_config, "architectures", None)
|
65 |
if architectures:
|
66 |
architecture = ";".join(architectures)
|
67 |
-
if model_type == ModelType.Unknown:
|
68 |
-
model_type = ModelType.from_str(getattr(model_config, "model_type", "Unknown"))
|
69 |
|
70 |
# Extract results available in this file (some results are split in several files)
|
71 |
results = {}
|
|
|
64 |
architectures = getattr(model_config, "architectures", None)
|
65 |
if architectures:
|
66 |
architecture = ";".join(architectures)
|
|
|
|
|
67 |
|
68 |
# Extract results available in this file (some results are split in several files)
|
69 |
results = {}
|