diff --git a/opencompass/partitioners/base.py b/opencompass/partitioners/base.py index 199728c6..17cedfc7 100644 --- a/opencompass/partitioners/base.py +++ b/opencompass/partitioners/base.py @@ -102,7 +102,7 @@ class BasePartitioner: return tasks def parse_model_dataset_args(self, cfg: ConfigDict): - #breakpoint() + models = cfg['models'] datasets = cfg['datasets']