from mmengine.config import read_base with read_base(): from .datasets.demo.demo_gsm8k_base_gen import gsm8k_datasets from .datasets.demo.demo_math_base_gen import math_datasets from .models.qwen.hf_qwen2_1_5b import models as hf_qwen2_1_5b_models from .models.hf_internlm.hf_internlm2_1_8b import models as hf_internlm2_1_8b_models datasets = gsm8k_datasets + math_datasets models = hf_qwen2_1_5b_models + hf_internlm2_1_8b_models