add readme (#10)
* Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * Update Readme.md * remove submodule * add mPLUG MiniGPT4 * Update Readme.md * Update Readme.md * Update Readme.md --------- Co-authored-by: Yuliang Liu <34134635+Yuliang-Liu@users.noreply.github.com>
This commit is contained in:
33
models/MiniGPT4/minigpt4/processors/__init__.py
Normal file
33
models/MiniGPT4/minigpt4/processors/__init__.py
Normal file
@@ -0,0 +1,33 @@
|
||||
"""
|
||||
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
|
Reference in New Issue
Block a user