-
Notifications
You must be signed in to change notification settings - Fork 1.1k
Description
Describe the bug
It occured at unstructured.partition.pdf(), when strategy=hi_res,
if strategy=fast, it unable to extract from pdf.
TypeError: unstructured_inference.models.yolox.UnstructuredYoloXModel.initialize() argument after ** must be a mapping, not LazyDict
elements = partition_pdf(
filename=file_path,
strategy=strategy,
infer_table_structure=True,
hi_res_model_name=model_name,
languages=['chi_sim']
)
Traceback (most recent call last):
File "/root/autodl-tmp/aicode/hw-chat-0.1-jry/data/parse/unstructured_processor.py", line 38, in
lambda: self.process_file(file_path, strategy, model_name),
File "/root/autodl-tmp/aicode/hw-chat-0.1-jry/data/parse/unstructured_processor.py", line 133, in process_file
elements = partition_pdf(
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/documents/elements.py", line 585, in wrapper
elements = func(*args, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/file_utils/filetype.py", line 816, in wrapper
elements = func(*args, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/file_utils/filetype.py", line 774, in wrapper
elements = func(*args, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/chunking/dispatch.py", line 74, in wrapper
elements = func(*args, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/partition/pdf.py", line 228, in partition_pdf
return partition_pdf_or_image(
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/partition/pdf.py", line 341, in partition_pdf_or_image
elements = _partition_pdf_or_image_local(
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/utils.py", line 216, in wrapper
return func(*args, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured/partition/pdf.py", line 646, in _partition_pdf_or_image_local
inferred_document_layout = process_file_with_model(
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured_inference/inference/layout.py", line 371, in process_file_with_model
model = get_model(model_name, **kwargs)
File "/root/miniconda3/envs/hw-chat-npu/lib/python3.10/site-packages/unstructured_inference/models/base.py", line 97, in get_model
model.initialize(**initialize_params)
TypeError: unstructured_inference.models.yolox.UnstructuredYoloXModel.initialize() argument after ** must be a mapping, not LazyDict
It seems to be a initialize yolox model failed. How can i fix this?