使用python PPOCRLabel.py --lang ch运行加载PP-Chart2Table模型出现报错 #220
Description
Traceback (most recent call last):
File "D:\PPOCRLabel-main\PPOCRLabel.py", line 3852, in
sys.exit(main())
File "D:\PPOCRLabel-main\PPOCRLabel.py", line 3847, in main
app, _win = get_main_app(sys.argv)
File "D:\PPOCRLabel-main\PPOCRLabel.py", line 3828, in get_main_app
win = MainWindow(
File "D:\PPOCRLabel-main\PPOCRLabel.py", line 223, in init
self.table_ocr = PPStructureV3(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddleocr_pipelines\pp_structurev3.py", line 137, in init
super().init(**kwargs)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddleocr_pipelines\base.py", line 67, in init
self.paddlex_pipeline = self._create_paddlex_pipeline()
File "D:\Anaconda\envs\ocr\lib\site-packages\paddleocr_pipelines\base.py", line 105, in _create_paddlex_pipeline
return create_pipeline(config=self.merged_paddlex_config, **kwargs)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines_init.py", line 167, in create_pipeline
pipeline = BasePipeline.get(pipeline_name)(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\utils\deps.py", line 206, in _wrapper
return old_init_func(self, *args, **kwargs)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines_parallel.py", line 103, in init
self._pipeline = self._create_internal_pipeline(config, self.device)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines_parallel.py", line 158, in _create_internal_pipeline
return self.pipeline_cls(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines\layout_parsing\pipeline_v2.py", line 84, in init
self.inintial_predictor(config)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines\layout_parsing\pipeline_v2.py", line 215, in inintial_predictor
self.chart_recognition_model = self.create_model(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\pipelines\base.py", line 106, in create_model
model = create_predictor(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\models_init.py", line 87, in create_predictor
return BasePredictor.get(model_name)(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\models\doc_vlm\predictor.py", line 66, in init
self.infer, self.processor = self._build(**kwargs)
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\models\doc_vlm\predictor.py", line 131, in _build
model = PPChart2TableInference.from_pretrained(
File "D:\Anaconda\envs\ocr\lib\site-packages\paddlex\inference\models\common\vlm\transformers\model_utils.py", line 1940, in from_pretrained
transpose_weight_keys=transpose_weight_keys,
UnboundLocalError: local variable 'transpose_weight_keys' referenced before assignment
通过更改model_utils.py解决
transpose_weight_keys = None # 安全初始化
try:
transpose_weight_keys = model.get_transpose_weight_keys()
except NotImplementedError:
if convert_from_hf:
raise ValueError("convert_from_hf=True is not supported")