diff --git a/demo/demo_vllm.py b/demo/demo_vllm.py index dc3a73d..3b6f4f8 100755 --- a/demo/demo_vllm.py +++ b/demo/demo_vllm.py @@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm parser = argparse.ArgumentParser() parser.add_argument("--ip", type=str, default="localhost") 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("--prompt_mode", type=str, default="prompt_layout_all_en",help=( "Choose a task prompt: " diff --git a/demo/demo_vllm_general.py b/demo/demo_vllm_general.py index 773e2c3..d230c57 100755 --- a/demo/demo_vllm_general.py +++ b/demo/demo_vllm_general.py @@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm parser = argparse.ArgumentParser() parser.add_argument("--ip", type=str, default="localhost") 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.") args = parser.parse_args() diff --git a/demo/demo_vllm_svg.py b/demo/demo_vllm_svg.py index 83104ad..e7719b6 100755 --- a/demo/demo_vllm_svg.py +++ b/demo/demo_vllm_svg.py @@ -10,7 +10,7 @@ from dots_ocr.model.inference import inference_with_vllm parser = argparse.ArgumentParser() parser.add_argument("--ip", type=str, default="localhost") 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") args = parser.parse_args()