FAPM / lavis /tasks /text_to_image_generation.py
wenkai's picture
Upload 560 files
4b532c0 verified
raw
history blame
609 Bytes
"""
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)