Spaces:
Running
on
Zero
Running
on
Zero
""" | |
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.common.registry import registry | |
__all__ = [ | |
"BaseProcessor", | |
"Blip2ImageTrainProcessor", | |
"Blip2ImageEvalProcessor", | |
"BlipCaptionProcessor", | |
] | |
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 | |