Joschka Strueber commited on
Commit
ec5f717
·
1 Parent(s): 9f3c166

[Add] only load cached models

Browse files
Files changed (1) hide show
  1. 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 get_leaderboard_models():
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()