Spaces:
Paused
Paused
Added revision + increased batch size
Browse files
main_backend_lighteval.py
CHANGED
@@ -80,7 +80,7 @@ def run_auto_eval():
|
|
80 |
eval_request=eval_request,
|
81 |
task_names=TASKS_LIGHTEVAL,
|
82 |
local_dir=EVAL_RESULTS_PATH_BACKEND,
|
83 |
-
batch_size=
|
84 |
accelerator=ACCELERATOR,
|
85 |
region=REGION,
|
86 |
vendor=VENDOR,
|
|
|
80 |
eval_request=eval_request,
|
81 |
task_names=TASKS_LIGHTEVAL,
|
82 |
local_dir=EVAL_RESULTS_PATH_BACKEND,
|
83 |
+
batch_size=25,
|
84 |
accelerator=ACCELERATOR,
|
85 |
region=REGION,
|
86 |
vendor=VENDOR,
|
src/backend/run_eval_suite_lighteval.py
CHANGED
@@ -23,6 +23,7 @@ def run_evaluation(eval_request: EvalRequest, task_names: str, batch_size: int,
|
|
23 |
args = DefaultNamespace(**{
|
24 |
"endpoint_model_name": eval_request.model,
|
25 |
"model_dtype": eval_request.precision,
|
|
|
26 |
"accelerator": accelerator,
|
27 |
"vendor": vendor,
|
28 |
"region": region,
|
@@ -42,7 +43,7 @@ def run_evaluation(eval_request: EvalRequest, task_names: str, batch_size: int,
|
|
42 |
"tasks": task_names,
|
43 |
"dataset_loading_processes": 24,
|
44 |
"num_fewshot_seeds": 0,
|
45 |
-
"reuse_existing": False
|
46 |
})
|
47 |
|
48 |
try:
|
|
|
23 |
args = DefaultNamespace(**{
|
24 |
"endpoint_model_name": eval_request.model,
|
25 |
"model_dtype": eval_request.precision,
|
26 |
+
"revision": eval_request.revision,
|
27 |
"accelerator": accelerator,
|
28 |
"vendor": vendor,
|
29 |
"region": region,
|
|
|
43 |
"tasks": task_names,
|
44 |
"dataset_loading_processes": 24,
|
45 |
"num_fewshot_seeds": 0,
|
46 |
+
"reuse_existing": False,
|
47 |
})
|
48 |
|
49 |
try:
|