"""
 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.processors.base_processor import BaseProcessor
# from minigpt4.processors.blip_processors import (
#     Blip2ImageTrainProcessor,
#     Blip2ImageEvalProcessor,
#     BlipCaptionProcessor,
# )
from minigpt4.processors.imagebind_processor import (
    ImageBindCaptionProcessor,
    ImageBindVisionTrainProcessor,
    ImageBindVisionEvalProcessor
)

from minigpt4.common.registry import registry

__all__ = [
    "BaseProcessor",
    # "Blip2ImageTrainProcessor",
    # "Blip2ImageEvalProcessor",
    # "BlipCaptionProcessor",
    "ImageBindCaptionProcessor",
    "ImageBindVisionTrainProcessor",
    "ImageBindVisionEvalProcessor"
]


def load_processor(name, cfg=None):
    """
    Example

    >>> processor = load_processor("alpro_video_train", cfg=None)
    """
    processor = registry.get_processor_class(name).from_config(cfg)

    return processor