Unverified Commit 9e42cb16 authored by Songyang Zhang's avatar Songyang Zhang Committed by GitHub
Browse files

[Feature] Update xunfei api (#572)

* update xunfei api

* fix lint

* avoid warning
parent 7f77e8da
...@@ -247,6 +247,8 @@ class XunFei(BaseAPIModel): ...@@ -247,6 +247,8 @@ class XunFei(BaseAPIModel):
self.release() self.release()
if err_code == 0: if err_code == 0:
return msg.strip() return msg.strip()
if err_code == 10014: # skip safety problem
return 'None'
if err_code == 10013: if err_code == 10013:
return err_data['header']['message'] return err_data['header']['message']
......
...@@ -15,7 +15,6 @@ from mmengine.dist import init_dist ...@@ -15,7 +15,6 @@ from mmengine.dist import init_dist
from mmengine.evaluator import Evaluator from mmengine.evaluator import Evaluator
from mmengine.logging import print_log from mmengine.logging import print_log
from mmengine.model.wrappers import MMDistributedDataParallel from mmengine.model.wrappers import MMDistributedDataParallel
from mmengine.runner import Runner
from mmengine.utils import track_iter_progress from mmengine.utils import track_iter_progress
from opencompass.registry import MM_MODELS, TASKS from opencompass.registry import MM_MODELS, TASKS
...@@ -115,6 +114,8 @@ class MultimodalInferTask: ...@@ -115,6 +114,8 @@ class MultimodalInferTask:
return template.format(task_cmd=command) return template.format(task_cmd=command)
def run(self): def run(self):
from mmengine.runner import Runner
# only support slurm, pytorch, mpi # only support slurm, pytorch, mpi
init_dist(self.cfg.launcher) init_dist(self.cfg.launcher)
self.logger.info(f'Task {self.name}') self.logger.info(f'Task {self.name}')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment