cunliangkong/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-03-22 09:53:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

add multimodal LLM BLIP-2 and InstructBLIP

Browse Source 
Former-commit-id: a730f89a972f1a9d37c718c716f199cb8d4903b2
This commit is contained in:
BUAADreamer
2024-04-23 18:45:43 +08:00
parent 1d2e372a8e
commit ab6dc0ea30
16 changed files with 710 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/data/__init__.py
View File

@@ -1,12 +1,12 @@
from .collator import PairwiseDataCollatorWithPadding
from .loader import get_dataset
from .loader import get_dataset, get_mm_dataset
from .template import Template, get_template_and_fix_tokenizer, templates
from .utils import Role, split_dataset
__all__ = [
"PairwiseDataCollatorWithPadding",
"get_dataset",
"get_mm_dataset",
"Template",
"get_template_and_fix_tokenizer",
"templates",