Spaces:
Runtime error
Runtime error
File size: 1,286 Bytes
e4bd7f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
"""
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 bubogpt.processors.base_processor import BaseProcessor
from bubogpt.processors.blip_processors import (
Blip2ImageTrainProcessor,
Blip2ImageEvalProcessor,
BlipCaptionProcessor,
)
from bubogpt.processors.imagebind_vision_processor import (
ImageBindCaptionProcessor,
ImageBindVisionTrainProcessor,
ImageBindVisionEvalProcessor
)
from bubogpt.processors.imagebind_audio_processor import (
ImageBindAudioTrainProcessor,
ImageBindAudioEvalProcessor,
)
from bubogpt.common.registry import registry
__all__ = [
"BaseProcessor",
"Blip2ImageTrainProcessor",
"Blip2ImageEvalProcessor",
"BlipCaptionProcessor",
"ImageBindCaptionProcessor",
"ImageBindVisionTrainProcessor",
"ImageBindVisionEvalProcessor",
"ImageBindAudioTrainProcessor",
"ImageBindAudioEvalProcessor",
]
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
|