Spaces:
Sleeping
Sleeping
Clémentine
commited on
Commit
·
8580b64
1
Parent(s):
8dafde0
*.py
Browse files- globals.py +1 -0
- utils/io.py +1 -1
- 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=
|
| 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=
|
| 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 |
)
|