Clémentine
commited on
Commit
•
6e79cea
1
Parent(s):
1d49827
fixes display for model params that are 0
Browse files
src/display_models/get_model_metadata.py
CHANGED
@@ -22,8 +22,8 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
|
|
22 |
model_info = api.model_info(model_name)
|
23 |
except huggingface_hub.utils._errors.RepositoryNotFoundError:
|
24 |
print("Repo not found!", model_name)
|
25 |
-
model_data[AutoEvalColumn.license.name] =
|
26 |
-
model_data[AutoEvalColumn.likes.name] =
|
27 |
model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
|
28 |
continue
|
29 |
|
@@ -36,7 +36,7 @@ def get_model_license(model_info):
|
|
36 |
try:
|
37 |
return model_info.cardData["license"]
|
38 |
except Exception:
|
39 |
-
return
|
40 |
|
41 |
|
42 |
def get_model_likes(model_info):
|
@@ -56,7 +56,7 @@ def get_model_size(model_name, model_info):
|
|
56 |
size = size_match.group(0)
|
57 |
return round(float(size[:-1]) if size[-1] == "b" else float(size[:-1]) / 1e3, 3)
|
58 |
except AttributeError:
|
59 |
-
return
|
60 |
|
61 |
|
62 |
def get_model_type(leaderboard_data: List[dict]):
|
|
|
22 |
model_info = api.model_info(model_name)
|
23 |
except huggingface_hub.utils._errors.RepositoryNotFoundError:
|
24 |
print("Repo not found!", model_name)
|
25 |
+
model_data[AutoEvalColumn.license.name] = "?"
|
26 |
+
model_data[AutoEvalColumn.likes.name] = 0
|
27 |
model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
|
28 |
continue
|
29 |
|
|
|
36 |
try:
|
37 |
return model_info.cardData["license"]
|
38 |
except Exception:
|
39 |
+
return "?"
|
40 |
|
41 |
|
42 |
def get_model_likes(model_info):
|
|
|
56 |
size = size_match.group(0)
|
57 |
return round(float(size[:-1]) if size[-1] == "b" else float(size[:-1]) / 1e3, 3)
|
58 |
except AttributeError:
|
59 |
+
return 0
|
60 |
|
61 |
|
62 |
def get_model_type(leaderboard_data: List[dict]):
|
src/display_models/read_results.py
CHANGED
@@ -27,7 +27,7 @@ class EvalResult:
|
|
27 |
results: dict
|
28 |
precision: str = ""
|
29 |
model_type: str = ""
|
30 |
-
weight_type: str = ""
|
31 |
date: str = ""
|
32 |
|
33 |
def to_dict(self):
|
|
|
27 |
results: dict
|
28 |
precision: str = ""
|
29 |
model_type: str = ""
|
30 |
+
weight_type: str = "Original"
|
31 |
date: str = ""
|
32 |
|
33 |
def to_dict(self):
|