|
"""
|
|
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)
|
|
|
