Nathan Habib
commited on
Commit
•
5da3025
1
Parent(s):
bc44e71
Fix merge
Browse files
src/display_models/get_model_metadata.py
CHANGED
@@ -26,7 +26,6 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
|
|
26 |
|
27 |
for model_data in tqdm(leaderboard_data):
|
28 |
model_name = model_data["model_name_for_query"]
|
29 |
-
<<<<<<< HEAD
|
30 |
|
31 |
if model_name in model_info_cache:
|
32 |
model_info = model_info_cache[model_name]
|
@@ -40,16 +39,6 @@ def get_model_infos_from_hub(leaderboard_data: List[dict]):
|
|
40 |
model_data[AutoEvalColumn.likes.name] = None
|
41 |
model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
|
42 |
continue
|
43 |
-
=======
|
44 |
-
try:
|
45 |
-
model_info = api.model_info(model_name)
|
46 |
-
except huggingface_hub.utils._errors.RepositoryNotFoundError:
|
47 |
-
print("Repo not found!", model_name)
|
48 |
-
model_data[AutoEvalColumn.license.name] = "?"
|
49 |
-
model_data[AutoEvalColumn.likes.name] = 0
|
50 |
-
model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
|
51 |
-
continue
|
52 |
-
>>>>>>> 6e79cea283b9033350b77806ca64c34a2e0cd323
|
53 |
|
54 |
model_data[AutoEvalColumn.license.name] = get_model_license(model_info)
|
55 |
model_data[AutoEvalColumn.likes.name] = get_model_likes(model_info)
|
|
|
26 |
|
27 |
for model_data in tqdm(leaderboard_data):
|
28 |
model_name = model_data["model_name_for_query"]
|
|
|
29 |
|
30 |
if model_name in model_info_cache:
|
31 |
model_info = model_info_cache[model_name]
|
|
|
39 |
model_data[AutoEvalColumn.likes.name] = None
|
40 |
model_data[AutoEvalColumn.params.name] = get_model_size(model_name, None)
|
41 |
continue
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
model_data[AutoEvalColumn.license.name] = get_model_license(model_info)
|
44 |
model_data[AutoEvalColumn.likes.name] = get_model_likes(model_info)
|