support train from scratch #4033 #4075

This commit is contained in:
hiyouga committed 2024-06-06 02:43:19 +08:00
1 parent 946f601136
commit a12a506c3d
2 files changed
+6

No files matched your search

+4
View File
@@ -101,6 +101,10 @@ class ModelArguments:
default=False,
metadata={"help": "Whether or not to upcast the output of lm_head in fp32."},
)
train_from_scratch: bool = field(
default=False,
metadata={"help": "Whether or not to randomly initialize the model weights."},
)
infer_backend: Literal["huggingface", "vllm"] = field(
default="huggingface",
metadata={"help": "Backend engine used at inference."},