Fix engine unit test (#1701)

2024-10-17 09:53:32 -07:00
parent 7feba41584
commit dd3809fad8
4 changed files with 12 additions and 8 deletions
--- a/benchmark/llava_bench/bench_sglang.py
+++ b/benchmark/llava_bench/bench_sglang.py
@@ -20,7 +20,7 @@ def image_qa(s, image_file, question):


 def main(args):
-    lines = read_jsonl(args.question_file)[: args.num_questions]
+    lines = list(read_jsonl(args.question_file))[: args.num_questions]
    arguments = [
        {
            "image_file": os.path.abspath(args.image_folder + "/" + l["image"]),