add rlhf-v dataset

This commit is contained in:
hiyouga committed 2024-09-01 22:57:41 +08:00
1 parent 55027282cd
commit 8e49940746
13 files changed
+118 -33

No files matched your search

+4
View File
@@ -138,6 +138,10 @@ class ModelArguments:
default=False,
metadata={"help": "Whether or not to randomly initialize the model weights."},
)
image_resolution: int = field(
default=512,
metadata={"help": "Keeps the height or width of image below this resolution."},
)
infer_backend: Literal["huggingface", "vllm"] = field(
default="huggingface",
metadata={"help": "Backend engine used at inference."},