Closed
Description
It seams that it is not possible anymore to install surya v0.13.1 because the Recognition model is broken.
Steps to reproduce
- Create an env that is completely new:
docker run -it --rm python:3.12 bash- Installing lib 0.13.1
pip install surya-ocr==0.13.1- Importing and loading model
python
from surya.recognition import RecognitionPredictor
recognition_predictor = RecognitionPredictor()
Leads to the following error:
>>> recognition_predictor = RecognitionPredictor()
Downloading text_recognition model...: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 9/9 [01:50<00:00, 12.24s/it]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python3.12/site-packages/surya/common/predictor.py", line 24, in __init__
self.model = loader.model(device, dtype)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/surya/recognition/loader.py", line 35, in model
config = SuryaOCRConfig.from_pretrained(self.checkpoint)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/surya/common/s3.py", line 132, in from_pretrained
return super().from_pretrained(local_path, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/transformers/configuration_utils.py", line 573, in from_pretrained
return cls.from_dict(config_dict, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/transformers/configuration_utils.py", line 770, in from_dict
logger.info(f"Model config {config}")
^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/transformers/configuration_utils.py", line 802, in __repr__
return f"{self.__class__.__name__} {self.to_json_string()}"
^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/transformers/configuration_utils.py", line 914, in to_json_string
config_dict = self.to_diff_dict()
^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/transformers/configuration_utils.py", line 822, in to_diff_dict
class_config_dict = self.__class__().to_dict() if not self.has_no_defaults_at_init else {}
^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.12/site-packages/surya/recognition/model/config.py", line 17, in __init__
encoder_config = kwargs.pop("encoder")
^^^^^^^^^^^^^^^^^^^^^
KeyError: 'encoder'
Metadata
Metadata
Assignees
Labels
No labels