meg-huggingface
commited on
Commit
•
0f1cbe4
1
Parent(s):
5cb2831
Changes to whatever
Browse files
main_backend_harness.py
CHANGED
@@ -72,7 +72,6 @@ def run_auto_eval():
|
|
72 |
results_repo=RESULTS_REPO,
|
73 |
batch_size='auto',
|
74 |
device=DEVICE,
|
75 |
-
no_cache=True,
|
76 |
limit=LIMIT
|
77 |
)
|
78 |
|
|
|
72 |
results_repo=RESULTS_REPO,
|
73 |
batch_size='auto',
|
74 |
device=DEVICE,
|
|
|
75 |
limit=LIMIT
|
76 |
)
|
77 |
|
src/backend/run_eval_suite_harness.py
CHANGED
@@ -26,7 +26,7 @@ class NumpyArrayEncoder(json.JSONEncoder):
|
|
26 |
else:
|
27 |
return super().default(obj)
|
28 |
|
29 |
-
def run_evaluation(eval_request: EvalRequest, task_names: list, num_fewshot: int, batch_size: int, device: str, local_dir: str, results_repo: str,
|
30 |
"""Runs one evaluation for the current evaluation request file, then pushes the results to the hub.
|
31 |
|
32 |
Args:
|
|
|
26 |
else:
|
27 |
return super().default(obj)
|
28 |
|
29 |
+
def run_evaluation(eval_request: EvalRequest, task_names: list, num_fewshot: int, batch_size: int or str, device: str, local_dir: str, results_repo: str, limit: int =None):
|
30 |
"""Runs one evaluation for the current evaluation request file, then pushes the results to the hub.
|
31 |
|
32 |
Args:
|