safetensors_rust.SafetensorError: framework paddle is invalid #4651
Description
Checklist:
- 查找历史相关issue寻求解答
- 翻阅FAQ
- 翻阅PaddleX 文档
- 确认bug是否在新版本里还未修复
描述问题
复现
-
您是否已经正常运行我们提供的教程？
-
您是否在教程的基础上修改代码内容？还请您提供运行的代码
-
您使用的数据集是？
-
请提供您出现的报错信息及相关log
环境
-
PaddlePaddle=3.0.0，PaddleX=3.3.4
-
Windows
-
Python=3.12
-
CUDA=12.9，cuDNN=9.14
File "F:\PaddleX\paddlex\inference\pipelines_init_.py", line 167, in create_pipeline
pipeline = BasePipeline.get(pipeline_name)(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\utils\deps.py", line 206, in _wrapper
return old_init_func(self, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\pipelines_parallel.py", line 103, in init
self._pipeline = self._create_internal_pipeline(config, self.device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\pipelines_parallel.py", line 158, in _create_internal_pipeline
return self.pipeline_cls(
^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\pipelines\paddleocr_vl\pipeline.py", line 127, in init
self.vl_rec_model = self.create_model(vl_rec_config)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\pipelines\base.py", line 106, in create_model
model = create_predictor(
^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\models_init.py", line 87, in create_predictor
return BasePredictor.get(model_name)(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\models\doc_vlm\predictor.py", line 66, in init
self.infer, self.processor = self._build(**kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\models\doc_vlm\predictor.py", line 152, in _build
model = PaddleOCRVLForConditionalGeneration.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\models\common\vlm\transformers\model_utils.py", line 1941, in from_pretrained
state_dict = load_state_dict(
^^^^^^^^^^^^^^^^
File "F:\PaddleX\paddlex\inference\models\common\vlm\transformers\model_utils.py", line 254, in load_state_dict
with safe_open(checkpoint_file, framework="paddle") as f:
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
safetensors_rust.SafetensorError: framework paddle is invalid
使用以下命令安装 safetensors 后，依然报错
python -m pip install https://paddle-whl.bj.bcebos.com/stable/cu129/safetensors/safetensors-0.6.2-cp38-abi3-win_amd64.whl