Fix engine unit test (#1701)
This commit is contained in:
@@ -17,7 +17,7 @@ pip3 install "torch>=2.1.2" "transformers>=4.36" pillow
|
||||
### Benchmark sglang
|
||||
Launch a server
|
||||
```
|
||||
python3 -m sglang.launch_server --model-path liuhaotian/llava-v1.5-7b --tokenizer-path llava-hf/llava-1.5-7b-hf --port 30000
|
||||
python3 -m sglang.launch_server --model-path liuhaotian/llava-v1.6-vicuna-7b --tokenizer-path llava-hf/llava-1.5-7b-hf --port 30000
|
||||
```
|
||||
|
||||
Run benchmark
|
||||
|
||||
@@ -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"]),
|
||||
|
||||
Reference in New Issue
Block a user