def infer(self, model_name, inputs, model_version, outputs, request_id, parameters)

函数功能

实现同步推理。

函数原型

def infer(self, model_name, inputs, model_version, outputs, request_id, parameters)

参数说明

参数名

参数类型

输入/输出

说明

model_name

str

输入

模型名称。

模型名称只支持由大小写字母、数字、中横线和下划线组成。

inputs

list

输入

模型输入。

目前只支持1个输入,shape取值为[1,n](n <= 16000),dataType只支持UINT32。

model_version

str

输入

模型版本,默认""。

非空时只支持由大小写字母、数字、中横线和下划线组成。

outputs

list

输入

指定需要返回的模型输出。如果为None则全部返回。目前只支持指定一个输出,name为“output0”。

request_id

str

输入

请求ID,默认""。

parameters

dict

输入

额外的请求参数,包括seed、temperature、top_k、top_p、do_sample和repetition_penalty,默认None。

seed

int64

输入

随机种子数。

取值(0,18446744073709551615],不传递随机指定。

temperature

float

输入

控制生成的随机性,较高的值会产生更多样化的输出。

取值范围大于0,默认值为1。

top_k

int32

输入

控制模型生成过程中考虑的词汇范围,只从概率最高的 k 个候选词中选择,0表示不做top_k。

取值范围:[0,2147483647],默认值为0。

top_p

float

输入

使用累计概率选择候选词,直到累计概率超过给定的阈值。

取值范围:(0,1],默认值为1。

do_sample

bool

输入

是否做sampling。

repetition_penalty

float

输入

用于减少在文本生成过程中出现重复片段的概率。

取值范围大于0,默认值为1.0。

返回值

Result对象表示同步推理结果。