video datasets

Former-commit-id: 8cafc7b055

src/llamafactory/extras/constants.py

@@ -95,6 +95,8 @@ SUPPORTED_CLASS_FOR_BLOCK_DIAG_ATTN = {
 
 SUPPORTED_CLASS_FOR_S2ATTN = {"llama"}
 
+VIDEO_PLACEHOLDER = "<video>"
+
 V_HEAD_WEIGHTS_NAME = "value_head.bin"
 
 V_HEAD_SAFE_WEIGHTS_NAME = "value_head.safetensors"

src/llamafactory/extras/packages.py

@@ -38,6 +38,10 @@ def _get_package_version(name: str) -> "Version":
         return version.parse("0.0.0")
 
 
+def is_pyav_available():
+    return _is_package_available("av")
+
+
 def is_fastapi_available():
     return _is_package_available("fastapi")