昇腾社区首页
中文
注册

img2img

功能描述

与大模型服务交互进行文生图,获取模型推理的结果。发送给大模型数据格式如下:

请求body数据格式如下:
{
"prompt": 生成图片提示词,
"image": 图片数据base64编码后的数据,
"size": 图片生成尺寸,
"model_name": 模型名字
}

函数原型

def img2img(prompt, image_content, size)

输入参数说明

参数名

数据类型

可选/必选

说明

prompt

str

必选

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

image_content

str

必选

图片数据base64编码后对应的字符串,长度取值范围:(0, 10 * 1024 * 1024]。

size

str

可选

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

返回值说明

数据类型

说明

dict

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