mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +00:00
27 lines
736 B
Python
27 lines
736 B
Python
|
"""
|
||
|
|
Copyright (c) 2022, salesforce.com, inc.
|
||
|
|
All rights reserved.
|
||
|
|
SPDX-License-Identifier: BSD-3-Clause
|
||
|
|
For full license text, see the LICENSE_Lavis file in the repo root or https://opensource.org/licenses/BSD-3-Clause
|
||
|
|
"""
|
||
|
|
|
||
|
|
from minigpt4.common.registry import registry
|
||
|
|
from minigpt4.tasks.base_task import BaseTask
|
||
|
|
from minigpt4.tasks.image_text_pretrain import ImageTextPretrainTask
|
||
|
|
|
||
|
|
|
||
|
|
def setup_task(cfg):
|
||
|
|
assert "task" in cfg.run_cfg, "Task name must be provided."
|
||
|
|
|
||
|
|
task_name = cfg.run_cfg.task
|
||
|
|
task = registry.get_task_class(task_name).setup_task(cfg=cfg)
|
||
|
|
assert task is not None, "Task {} not properly registered.".format(task_name)
|
||
|
|
|
||
|
|
return task
|
||
|
|
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"BaseTask",
|
||
|
|
"ImageTextPretrainTask",
|
||
|
|
]
|