support batch infer in vllm

2024-12-04 13:50:00 +00:00
parent dc78355002
commit 1324d158f9
29 changed files with 148 additions and 407 deletions
--- a/examples/inference/llava1_5.yaml
+++ b/examples/inference/llava1_5.yaml
@@ -1,2 +1,3 @@
 model_name_or_path: llava-hf/llava-1.5-7b-hf
 template: llava
+infer_backend: huggingface  # choices: [huggingface, vllm]