video datasets

Former-commit-id: 33f28ce82d9e44d2615909250dc56d6a4a03cd99
2024-09-05 02:04:17 +08:00
parent 2c1eef34cb
commit 1874d579c5
25 changed files with 359 additions and 148 deletions
--- a/src/llamafactory/webui/components/infer.py
+++ b/src/llamafactory/webui/components/infer.py
@@ -68,7 +68,7 @@ def create_infer_tab(engine: "Engine") -> Dict[str, "Component"]:
    engine.manager.get_elem_by_id("top.model_name").change(
        lambda model_name: gr.Column(visible=get_visual(model_name)),
        [engine.manager.get_elem_by_id("top.model_name")],
-        [chat_elems["image_box"]],
+        [chat_elems["mm_box"]],
    )

    return elem_dict