RuntimeError: (NotFound) Operator (one_hot) is not registered. [Hint: op_info_ptr should not be null.] (at ..\paddle/fluid/framework/op_info.h:152)

[Idaydayup]

代码

import gradio as gr from paddlenlp import Taskflow import numpy as np from PIL import Image import uuid

初始化文档智能任务模型

docprompt = Taskflow("document_intelligence")

定义模型推理函数

def model_inference(image, prompt): prompt = prompt.split("?")[:-1] img = Image.fromarray(np.uint8(image)) filename = "image/" + str(uuid.uuid4()) + ".png" img.save(filename) res = docprompt([{"doc": filename, "prompt": prompt}]) json_out = {"result": res} print(json_out) return image, json_out

定义清除函数

def clear_all(): return None, None, None

使用Gradio构建用户界面

with gr.Blocks() as demo: gr.Markdown("ERNIE-Layout") with gr.Column(scale=1, min_width=100):

输入图片和文本框

    img_in = gr.Image(value="图片.jpg", label="Input")
    text = gr.Textbox(value="发票号码是多少?校验码是多少?", label="输入问题:", lines=2)
    with gr.Row():
        # 清除和提交按钮
        btn1 = gr.Button("Clear")
        btn2 = gr.Button("Submit")
        json_out = gr.JSON(label="Information Extraction Output")
        img_out = gr.Image(label="Output")  # 输出图片
        # 绑定按钮点击事件到函数
        btn1.click(fn=clear_all, inputs=None, outputs=[img_in, img_out, json_out])
        btn2.click(fn=model_inference, inputs=[img_in, text], outputs=[img_out, json_out])

demo.launch(server_port=7008) # 启动Gradio界面，监听7008端口)

报错D:\anaconda\envs\p5\python.exe D:\pycharm\实训2\任务5\内容识别.py

D:\anaconda\envs\p5\lib\site-packages_distutils_hack__init.py:33: UserWarning: Setuptools is replacing distutils. warnings.warn("Setuptools is replacing distutils.") E1228 17:47:01.441207 15760 analysis_config.cc:653] Please compile with MKLDNN first to use MKLDNN Traceback (most recent call last): File "D:\pycharm\实训2\任务5\内容识别.py", line 18, in docprompt = Taskflow("document_intelligence") File "D:\anaconda\envs\p5\lib\site-packages\paddlenlp\taskflow\taskflow.py", line 804, in init__ self.task_instance = task_class( File "D:\anaconda\envs\p5\lib\site-packages\paddlenlp\taskflow\document_intelligence.py", line 68, in init self._get_inference_model() File "D:\anaconda\envs\p5\lib\site-packages\paddlenlp\taskflow\task.py", line 372, in _get_inference_model self._prepare_static_mode() File "D:\anaconda\envs\p5\lib\site-packages\paddlenlp\taskflow\task.py", line 227, in _prepare_static_mode self.predictor = paddle.inference.create_predictor(self._config) RuntimeError: (NotFound) Operator (one_hot) is not registered. [Hint: op_info_ptr should not be null.] (at ..\paddle/fluid/framework/op_info.h:152)

进程已结束,退出代码1

[likenamehaojie]

我也遇到了类似的问题。

[m7mdhka]

+1 same issue

[magicianCoder]

应该是paddle版本导致，退回到paddlepaddle==1.5.0就好了（参照taskflow无法使用document_intelligent）

[QQbbla]

回退倒2.5之后会出现别的error，pipe = DocPipeline(preprocessor=preprocessor, docreader=docprompter) TypeError: init() got an unexpected keyword argument 'docreader' how to solve it?

[jonyllamado]

Try downgrading paddlepaddle-gpu from 2.6.2 to 2.5.2

python3 -m pip install paddlepaddle-gpu==2.5.2.post120 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html