support llava video (#426)

2024-05-14 07:57:00 +08:00
parent 5dc55a5f02
commit 0992d85f92
37 changed files with 1139 additions and 222 deletions
--- a/examples/quick_start/srt_example_llava.py
+++ b/examples/quick_start/srt_example_llava.py
@@ -14,7 +14,7 @@ def single():
    state = image_qa.run(
        image_path="images/cat.jpeg",
        question="What is this?",
-        max_new_tokens=64)
+        max_new_tokens=128)
    print(state["answer"], "\n")


@@ -36,7 +36,7 @@ def batch():
            {"image_path": "images/cat.jpeg", "question":"What is this?"},
            {"image_path": "images/dog.jpeg", "question":"What is this?"},
        ],
-        max_new_tokens=64,
+        max_new_tokens=128,
    )
    for s in states:
        print(s["answer"], "\n")