Spaces:
Running
Running
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")
|