Spaces:
Sleeping
Sleeping
Joschka Strueber
commited on
Commit
·
ec5f717
1
Parent(s):
9f3c166
[Add] only load cached models
Browse files- src/dataloading.py +9 -1
src/dataloading.py
CHANGED
|
@@ -5,7 +5,7 @@ from huggingface_hub import HfApi
|
|
| 5 |
from functools import lru_cache
|
| 6 |
|
| 7 |
|
| 8 |
-
def
|
| 9 |
api = HfApi()
|
| 10 |
|
| 11 |
# Load prechecked models
|
|
@@ -54,6 +54,14 @@ def get_leaderboard_models():
|
|
| 54 |
return sorted(models)
|
| 55 |
|
| 56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
@lru_cache(maxsize=1)
|
| 58 |
def get_leaderboard_models_cached():
|
| 59 |
return get_leaderboard_models()
|
|
|
|
| 5 |
from functools import lru_cache
|
| 6 |
|
| 7 |
|
| 8 |
+
def get_leaderboard_models_reload():
|
| 9 |
api = HfApi()
|
| 10 |
|
| 11 |
# Load prechecked models
|
|
|
|
| 54 |
return sorted(models)
|
| 55 |
|
| 56 |
|
| 57 |
+
def get_leaderboard_models():
|
| 58 |
+
# Load prechecked (ungated) models
|
| 59 |
+
with open("models.txt", "r") as f:
|
| 60 |
+
ungated_models = [line.strip() for line in f]
|
| 61 |
+
|
| 62 |
+
return sorted(ungated_models)
|
| 63 |
+
|
| 64 |
+
|
| 65 |
@lru_cache(maxsize=1)
|
| 66 |
def get_leaderboard_models_cached():
|
| 67 |
return get_leaderboard_models()
|