mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +00:00
19 lines
538 B
Python
19 lines
538 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
|
||
|
|
|
||
|
|
|
||
|
|
@registry.register_task("image_text_pretrain")
|
||
|
|
class ImageTextPretrainTask(BaseTask):
|
||
|
|
def __init__(self):
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
def evaluation(self, model, data_loader, cuda_enabled=True):
|
||
|
|
pass
|