Shaltiel commited on
Commit
7135a84
·
1 Parent(s): 2e91fe1

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=1,
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: