实现同步推理。
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对象表示同步推理结果。