diff --git a/xinference/model/image/model_spec.json b/xinference/model/image/model_spec.json index 51924f0fa6..40fede5f5c 100644 --- a/xinference/model/image/model_spec.json +++ b/xinference/model/image/model_spec.json @@ -1685,5 +1685,33 @@ }, "featured": false, "updated_at": 1770994886 + }, + { + "version": 2, + "model_name": "PaddleOCR-VL-1.6", + "model_family": "ocr", + "model_ability": [ + "ocr" + ], + "virtualenv": { + "packages": [ + "pillow==11.3.0", + "transformers==4.55.0", + "#system_numpy#", + "#system_torch#" + ] + }, + "model_src": { + "huggingface": { + "model_id": "PaddlePaddle/PaddleOCR-VL-1.6", + "model_revision": "main" + }, + "modelscope": { + "model_id": "PaddlePaddle/PaddleOCR-VL-1.6", + "model_revision": "master" + } + }, + "featured": false, + "updated_at": 1781512046 } ] diff --git a/xinference/model/image/ocr/paddleocr_vl.py b/xinference/model/image/ocr/paddleocr_vl.py index 9a5c580c46..b202510ceb 100644 --- a/xinference/model/image/ocr/paddleocr_vl.py +++ b/xinference/model/image/ocr/paddleocr_vl.py @@ -33,7 +33,8 @@ class PaddleOCRVLModel(OCRModel): @classmethod def match(cls, model_family: "ImageModelFamilyV2") -> bool: - return model_family.model_name == "PaddleOCR-VL" + model_name = model_family.model_name + return model_name == "PaddleOCR-VL" or model_name.startswith("PaddleOCR-VL-") def __init__( self,