Spaces:
Sleeping
Sleeping
Joschka Strueber
commited on
Commit
·
81438ca
1
Parent(s):
047f32f
[Fix] add check if cached files have been saved
Browse files- src/dataloading.py +4 -1
src/dataloading.py
CHANGED
|
@@ -10,7 +10,10 @@ def get_leaderboard_models():
|
|
| 10 |
api = HfApi()
|
| 11 |
|
| 12 |
# Load prechecked models
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# List all datasets in the open-llm-leaderboard organization
|
| 16 |
dataset_list = api.list_datasets(author="open-llm-leaderboard")
|
|
|
|
| 10 |
api = HfApi()
|
| 11 |
|
| 12 |
# Load prechecked models
|
| 13 |
+
try:
|
| 14 |
+
ungated_models = set(line.strip() for line in open("models.txt"))
|
| 15 |
+
except FileNotFoundError:
|
| 16 |
+
ungated_models = set()
|
| 17 |
|
| 18 |
# List all datasets in the open-llm-leaderboard organization
|
| 19 |
dataset_list = api.list_datasets(author="open-llm-leaderboard")
|