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

force register (#1311)

Browse Source
This commit is contained in:
Fengzhe Zhou 2024-07-11 19:59:35 +08:00 committed by GitHub
parent 889e7e1140
commit 62f55987f1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/tasks/openicl_attack.py
View File

@ -15,7 +15,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
task_abbr_from_cfg)
@TASKS.register_module(force=(__name__ == '__main__')) # A hack for script run
@TASKS.register_module()
class OpenICLAttackTask(BaseTask):
"""OpenICL Inference Task.

2
opencompass/tasks/openicl_eval.py
View File

@ -55,7 +55,7 @@ def extract_role_pred(s: str, begin_str: Optional[str],
return s[start:end]
@TASKS.register_module(force=(__name__ == '__main__')) # A hack for script run
@TASKS.register_module()
class OpenICLEvalTask(BaseTask):
"""OpenICL Evaluation Task.

2
opencompass/tasks/openicl_infer.py
View File

@ -17,7 +17,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
model_abbr_from_cfg, task_abbr_from_cfg)
@TASKS.register_module(force=(__name__ == '__main__')) # A hack for script run
@TASKS.register_module()
class OpenICLInferTask(BaseTask):
"""OpenICL Inference Task.