From 57e98f30bb56d1b9e61bca9beb0131ccb44d1e39 Mon Sep 17 00:00:00 2001 From: zhangsongyang Date: Fri, 25 Apr 2025 09:43:27 +0000 Subject: [PATCH] Update --- opencompass/utils/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opencompass/utils/run.py b/opencompass/utils/run.py index 31cabacd..0a5fb0e0 100644 --- a/opencompass/utils/run.py +++ b/opencompass/utils/run.py @@ -313,7 +313,7 @@ def change_accelerator(models, accelerator): model_kwargs=model_kwargs, max_seq_len=model.get('max_seq_len', None), max_out_len=model['max_out_len'], - batch_size=model['batch_size'], + batch_size=model.get('batch_size', 16), run_cfg=model['run_cfg'], stop_words=model.get('stop_words', []), )