昇腾社区首页
中文
注册

text2img

功能描述

与SD服务交互进行文生图,获取SD模型推理的结果。

请求body数据格式如下:

{
"prompt": 文本生成提示词,
"output_format": 生成的图片格式,
"size": 生成图片尺寸,
"model_name": 模型名字
}

函数原型

def text2img(prompt, output_format, size)

输入参数说明

参数名

数据类型

可选/必选

说明

prompt

str

必选

生成图片的提示词。长度取值范围[1, 1024 * 1024]。

output_format

str

可选

生成图片的格式。取值类型png、jpeg、jpg和webp,默认值为“png”

size

str

可选

图片生成尺寸,表示为"height*width",具体支持的尺寸由对应的大模型决定,正则匹配格式为: "^\d{1,5}\*\d{1,5}$",默认"512*512",当前支持的模型生成的图片尺寸为"512 * 512"。

返回值说明

数据类型

说明

dict

返回格式为{"prompt": prompt, "result": data},其中prompt为图片生成的提示词,result为大模型推理结果的图片base64编码后的数据。