| """
|
| Copyright (c) 2022, salesforce.com, inc.
|
| All rights reserved.
|
| SPDX-License-Identifier: BSD-3-Clause
|
| For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
|
| """
|
|
|
| from lavis.tasks import BaseTask
|
| from lavis.common.registry import registry
|
|
|
|
|
| @registry.register_task("text-to-image-generation")
|
| class TextToImageGenerationTask(BaseTask):
|
| def __init__(self, cfg):
|
| super().__init__()
|
|
|
| self.cfg = cfg
|
|
|
| @classmethod
|
| def setup_task(cls, cfg):
|
| run_cfg = cfg.run_cfg
|
|
|
| return cls(cfg=run_cfg)
|
|
|
