jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
zhangsongyang 2025-04-25 09:43:27 +00:00
parent 45af358798
commit 57e98f30bb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/utils/run.py
View File

@ -313,7 +313,7 @@ def change_accelerator(models, accelerator):
model_kwargs=model_kwargs, model_kwargs=model_kwargs,
max_seq_len=model.get('max_seq_len', None), max_seq_len=model.get('max_seq_len', None),
max_out_len=model['max_out_len'], max_out_len=model['max_out_len'],
batch_size=model['batch_size'], batch_size=model.get('batch_size', 16),
run_cfg=model['run_cfg'], run_cfg=model['run_cfg'],
stop_words=model.get('stop_words', []), stop_words=model.get('stop_words', []),
) )