From 9e42cb163bc30a24d61bf92335467894357e1f8c Mon Sep 17 00:00:00 2001 From: Songyang Zhang Date: Fri, 10 Nov 2023 22:46:06 +0800 Subject: [PATCH] [Feature] Update xunfei api (#572) * update xunfei api * fix lint * avoid warning --- opencompass/models/xunfei_api.py | 2 ++ opencompass/tasks/mm_infer.py | 3 ++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/opencompass/models/xunfei_api.py b/opencompass/models/xunfei_api.py index 9ff4780d..72f3815f 100644 --- a/opencompass/models/xunfei_api.py +++ b/opencompass/models/xunfei_api.py @@ -247,6 +247,8 @@ class XunFei(BaseAPIModel): self.release() if err_code == 0: return msg.strip() + if err_code == 10014: # skip safety problem + return 'None' if err_code == 10013: return err_data['header']['message'] diff --git a/opencompass/tasks/mm_infer.py b/opencompass/tasks/mm_infer.py index d46c59b2..51c98fac 100644 --- a/opencompass/tasks/mm_infer.py +++ b/opencompass/tasks/mm_infer.py @@ -15,7 +15,6 @@ from mmengine.dist import init_dist from mmengine.evaluator import Evaluator from mmengine.logging import print_log from mmengine.model.wrappers import MMDistributedDataParallel -from mmengine.runner import Runner from mmengine.utils import track_iter_progress from opencompass.registry import MM_MODELS, TASKS @@ -115,6 +114,8 @@ class MultimodalInferTask: return template.format(task_cmd=command) def run(self): + from mmengine.runner import Runner + # only support slurm, pytorch, mpi init_dist(self.cfg.launcher) self.logger.info(f'Task {self.name}')