add multimodal LLM BLIP-2 and InstructBLIP

This commit is contained in:
BUAADreamer
2024-04-23 18:45:43 +08:00
parent 5722ada12b
commit 4dcb11eab7
18 changed files with 4982 additions and 39 deletions

View File

@@ -1,10 +1,11 @@
from .loader import load_model, load_tokenizer
from .loader import load_model, load_tokenizer, load_processor, load_mm_model
from .utils import find_all_linear_modules, load_valuehead_params
__all__ = [
"load_model",
"load_mm_model",
"load_tokenizer",
"load_processor",
"load_valuehead_params",
"find_all_linear_modules",
]