merge model part to the text stream

This commit is contained in:
BUAADreamer
2024-04-25 08:20:41 +08:00
parent 8239907f57
commit 838eb87a96
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