We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d0aa341 commit ccd73f2Copy full SHA for ccd73f2
tests/e2e/long_term/test_deepseek_v2_lite_tp2_accuracy.py
@@ -38,7 +38,7 @@
38
39
40
def run_test(model_name, queue, more_args=None):
41
- model_args = f"pretrained={model_name},max_model_len=4096,trust_remote_code=True,tensor_parallel_size=4"
+ model_args = f"pretrained={model_name},max_model_len=4096,trust_remote_code=True,tensor_parallel_size=4,enforce_eager=True"
42
if more_args is not None:
43
model_args = f"{model_args},{more_args}"
44
results = lm_eval.simple_evaluate(
0 commit comments