merge model part to the text stream

Former-commit-id: 838eb87a96
This commit is contained in:
BUAADreamer
2024-04-25 08:20:41 +08:00
parent cb1c66a810
commit 4e032ff95e
5 changed files with 24 additions and 172 deletions

View File

@@ -169,6 +169,10 @@ class ModelArguments:
default=False,
metadata={"help": "For debugging purposes, print the status of the parameters in the model."},
)
use_mllm: bool = field(
default=False,
metadata={"help": "Whether use Multimodal LLM."},
)
def __post_init__(self):
self.compute_dtype = None