jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Feature] Update xunfei api (#572)

Browse Source 
* update xunfei api

* fix lint

* avoid warning
This commit is contained in:
Songyang Zhang 2023-11-10 22:46:06 +08:00 committed by GitHub
parent 7f77e8dae5
commit 9e42cb163b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/models/xunfei_api.py
View File

@ -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']

3
opencompass/tasks/mm_infer.py
View File

@ -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:
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}')