add mindspore modelnotfound error

Author: limuan

ais_bench/benchmark/configs/models/mf_models/mf_model.py

@@ -1,4 +1,4 @@
-from ais_bench.benchmark.models import MindFormerModel
+from ais_bench.benchmark.models.local_models.mindformers_model import MindFormerModel
 
 models = [
     dict(

ais_bench/benchmark/models/__init__.py

@@ -15,4 +15,3 @@
 from ais_bench.benchmark.models.api_models.tgi_api import TGICustomAPIStream # noqa: F401
 from ais_bench.benchmark.models.api_models.vllm_custom_api_chat import VllmMultiturnAPIChatStream # noqa: F401
 from ais_bench.benchmark.models.local_models.vllm_offline_vl import VLLMOfflineVLModel
-from ais_bench.benchmark.models.local_models.mindformers_model import MindFormerModel

ais_bench/benchmark/models/local_models/mindformers_model.py

@@ -8,14 +8,26 @@
 from ais_bench.benchmark.models.local_models.base import BaseModel
 from ais_bench.benchmark.models import APITemplateParser
 from ais_bench.benchmark.registry import MODELS
-
-from mindspore import Tensor, Model
-from mindformers import  MindFormerConfig, build_context
-from mindformers.models import build_network
-from mindformers.core.parallel_config import build_parallel_config
-from mindformers.utils.load_checkpoint_utils import get_load_path_after_hf_convert
-from mindformers.trainer.utils import transform_and_load_checkpoint
-
+try:
+    from mindspore import Tensor, Model
+    from mindformers import  MindFormerConfig, build_context
+    from mindformers.models import build_network
+    from mindformers.core.parallel_config import build_parallel_config
+    from mindformers.utils.load_checkpoint_utils import get_load_path_after_hf_convert
+    from mindformers.trainer.utils import transform_and_load_checkpoint
+except ModuleNotFoundError as _mf_err:
+    MIND_FORMERS_INSTALL_GUIDE = (
+        "MindSpore/MindFormers is not available in the current environment. "
+        "To use `MindFormerModel`, install MindSpore and MindFormers and ensure any required "
+        "environment variables for your device are configured.\n\n"
+        "Quick steps (examples):\n"
+        "  1) Follow the official MindSpore installation guide for your platform: "
+        "https://www.mindspore.cn/install/en.\n"
+        "  2) Install MindFormers: `pip install mindformers` (or follow its docs); "
+        "alternatively, add a local MindFormers checkout to your PYTHONPATH, e.g.:\n"
+        "     PYTHONPATH=/path/to/mindformers:$PYTHONPATH\n"
+    )
+    raise ModuleNotFoundError(MIND_FORMERS_INSTALL_GUIDE) from _mf_err
 
 
 class MultiTokenEOSCriteria(transformers.StoppingCriteria):

ais_bench/benchmark/utils/config/build.py

@@ -162,7 +162,6 @@ def build_model_from_cfg(model_cfg: ConfigDict):
     model_cfg.pop("use_timestamp", None)
     model_cfg.pop("run_cfg", None)
     model_cfg.pop("request_rate", None)
-    batch_size = model_cfg.pop("batch_size", None)
     model_cfg.pop("abbr", None)
     model_cfg.pop("attr", None)
     model_cfg.pop("summarizer_abbr", None)