mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
a4b54048ae
* [Feature] Add Ruler datasets * pre-commit fixed * Add model specific tokenizer to dataset * pre-commit modified * remove unused import * fix linting * add trust_remote to tokenizer load * lint fix * comments resolved * fix lint * Add readme * Fix lint * ruler refactorize * fix lint * lint fix * updated * lint fix * fix wonderwords import issue * prompt modified * update * readme updated * update * ruler dataset added * Update --------- Co-authored-by: tonysy <sy.zhangbuaa@gmail.com>
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
from mmengine.config import read_base
|
|
|
|
with read_base():
|
|
from .groups.ruler import ruler_summary_groups
|
|
|
|
|
|
ruler_4k_summarizer = dict(
|
|
dataset_abbrs=['ruler_4k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
|
|
ruler_4k_summarizer = dict(
|
|
dataset_abbrs=['ruler_4k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
ruler_8k_summarizer = dict(
|
|
dataset_abbrs=['ruler_8k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
ruler_16k_summarizer = dict(
|
|
dataset_abbrs=['ruler_16k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
ruler_32k_summarizer = dict(
|
|
dataset_abbrs=['ruler_32k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
|
|
ruler_128k_summarizer = dict(
|
|
dataset_abbrs=['ruler_128k'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
|
|
ruler_1m_summarizer = dict(
|
|
dataset_abbrs=['ruler_1m'],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|
|
|
|
ruler_combined_summarizer = dict(
|
|
dataset_abbrs=[
|
|
'ruler_4k',
|
|
'ruler_8k',
|
|
'ruler_16k',
|
|
'ruler_32k',
|
|
'ruler_128k',
|
|
'ruler_1m',
|
|
],
|
|
summary_groups=sum(
|
|
[v for k, v in locals().items() if k.endswith('_summary_groups')], []
|
|
),
|
|
)
|