Clémentine commited on
Commit
8580b64
·
1 Parent(s): 8dafde0
Files changed (3) hide show
  1. globals.py +1 -0
  2. utils/io.py +1 -1
  3. utils/jobs.py +1 -1
globals.py CHANGED
@@ -11,6 +11,7 @@ job_results: Dict[str, JobResult] = {} # {model_provider_key: JobResult}
11
  results_lock: threading.Lock = threading.Lock()
12
 
13
  # Configuration
 
14
  RESULTS_DATASET_NAME: str = "IPTesting/inference-provider-test-results"
15
  LOCAL_CONFIG_FILE: str = "model_providers.txt"
16
  TASKS: str = "extended|ifeval|0,lighteval|gsm_plus|0,lighteval|gpqa:diamond|0"
 
11
  results_lock: threading.Lock = threading.Lock()
12
 
13
  # Configuration
14
+ NUM_MODELS_RUN: int = 10
15
  RESULTS_DATASET_NAME: str = "IPTesting/inference-provider-test-results"
16
  LOCAL_CONFIG_FILE: str = "model_providers.txt"
17
  TASKS: str = "extended|ifeval|0,lighteval|gsm_plus|0,lighteval|gpqa:diamond|0"
utils/io.py CHANGED
@@ -13,7 +13,7 @@ def get_models_providers() -> List[Tuple[str, List[str]]]:
13
  filter="text-generation",
14
  sort="likes",
15
  direction=-1,
16
- limit=1,
17
  expand="inferenceProviderMapping"
18
  )
19
 
 
13
  filter="text-generation",
14
  sort="likes",
15
  direction=-1,
16
+ limit=globals.NUM_MODELS_RUN,
17
  expand="inferenceProviderMapping"
18
  )
19
 
utils/jobs.py CHANGED
@@ -91,7 +91,7 @@ def run_single_job(model: str, provider: str, tasks: str) -> Optional[str]:
91
  "--results-org", "IPTesting",
92
  "--max-samples", "10"
93
  ],
94
- namespace="clefourrier",
95
  secrets={"HF_TOKEN": os.getenv("HF_TOKEN")},
96
  token=os.getenv("HF_TOKEN")
97
  )
 
91
  "--results-org", "IPTesting",
92
  "--max-samples", "10"
93
  ],
94
+ namespace=globals.NAMESPACE,
95
  secrets={"HF_TOKEN": os.getenv("HF_TOKEN")},
96
  token=os.getenv("HF_TOKEN")
97
  )