video datasets

Former-commit-id: 8cafc7b055
This commit is contained in:
hiyouga
2024-09-05 02:04:17 +08:00
parent 60d770e4b1
commit 9df7a26e6b
27 changed files with 420 additions and 148 deletions

View File

@@ -142,6 +142,10 @@ class ModelArguments:
default=512,
metadata={"help": "Keeps the height or width of image below this resolution."},
)
video_fps: float = field(
default=2.0,
metadata={"help": "The frames to sample per second for video training."},
)
infer_backend: Literal["huggingface", "vllm"] = field(
default="huggingface",
metadata={"help": "Backend engine used at inference."},