Mirrors/dots.ocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update vLLM demo scripts

Browse Source
This commit is contained in:
zhd36
2026-02-16 03:54:55 +08:00
parent 7124ee97ce
commit a0ae36da5a
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo_vllm.py
+1 -1
View File
@@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--ip", type=str, default="localhost") parser.add_argument("--ip", type=str, default="localhost")
parser.add_argument("--port", type=str, default="8000") parser.add_argument("--port", type=str, default="8000")
parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr") parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr-1.5")
parser.add_argument("--image_path", type=str, default="demo/demo_image1.jpg") parser.add_argument("--image_path", type=str, default="demo/demo_image1.jpg")
parser.add_argument("--prompt_mode", type=str, default="prompt_layout_all_en",help=( parser.add_argument("--prompt_mode", type=str, default="prompt_layout_all_en",help=(
"Choose a task prompt: " "Choose a task prompt: "
demo/demo_vllm_general.py
+1 -1
View File
@@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--ip", type=str, default="localhost") parser.add_argument("--ip", type=str, default="localhost")
parser.add_argument("--port", type=str, default="8000") parser.add_argument("--port", type=str, default="8000")
parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr") parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr-1.5")
parser.add_argument("--custom_prompt", type=str, default="Please describe the content of this image.") parser.add_argument("--custom_prompt", type=str, default="Please describe the content of this image.")
args = parser.parse_args() args = parser.parse_args()
demo/demo_vllm_svg.py
+1 -1
View File
@@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--ip", type=str, default="localhost") parser.add_argument("--ip", type=str, default="localhost")
parser.add_argument("--port", type=str, default="8000") parser.add_argument("--port", type=str, default="8000")
parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr") parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr-1.5-svg")
parser.add_argument("--prompt_mode", type=str, default="prompt_image_to_svg") parser.add_argument("--prompt_mode", type=str, default="prompt_image_to_svg")
args = parser.parse_args() args = parser.parse_args()