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

use example summarizer by default (#508)

Browse Source
This commit is contained in:
Fengzhe Zhou 2023-10-27 11:45:29 +08:00 committed by GitHub
parent 46e7ede938
commit 6405cd2db5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
opencompass/utils/run.py
View File

@ -99,10 +99,9 @@ def get_config_from_arg(args) -> Config:
run_cfg=dict(num_gpus=args.num_gpus)) run_cfg=dict(num_gpus=args.num_gpus))
models.append(model) models.append(model)
summarizer = {} summarizer = args.summarizer if args.summarizer is not None else 'example'
if args.summarizer:
summarizers_dir = os.path.join(args.config_dir, 'summarizers') summarizers_dir = os.path.join(args.config_dir, 'summarizers')
s = match_cfg_file(summarizers_dir, [args.summarizer])[0] s = match_cfg_file(summarizers_dir, [summarizer])[0]
get_logger().info(f'Loading {s[0]}: {s[1]}') get_logger().info(f'Loading {s[0]}: {s[1]}')
cfg = Config.fromfile(s[1]) cfg = Config.fromfile(s[1])
summarizer = cfg['summarizer'] summarizer = cfg['summarizer']